baal-org / baal

Bayesian active learning library for research and industrial usecases.
https://baal.readthedocs.io
Apache License 2.0
854 stars 84 forks source link

Different Loss Functions #278

Open TobiArndt opened 10 months ago

TobiArndt commented 10 months ago

Hi, I really like your work! I have got a question with respect to the loss functions: Is there a chance to use loss functions from the unsupervised regime? Like Student-Teacher Losses, where the final loss is the difference between the intermediate layer between a frozen teacher and a trainanble student network? This could be important for filtering anomalies from a dataset, so that only the non-anomaly images can be used for training the anomaly detector. Thank in advanced!

Dref360 commented 10 months ago

Hello, sorry for the delay.

Do you mean that you take the loss as a measure of uncertainty?