This library contains implementations of some histogram-based loss functions:
- Earth Mover Distrance Loss
- Histgramm Loss (paper, original code)
- Inverse Histogram Loss (our impovements)
- Bidirectinal Histogramm Loss (our impovements)
- Continuous Histogram Loss (paper)
Also there are implementations of another losses to compare:
- Negative Log-Likelihood
- Binomial Deviance loss (paper)
The instalation directly from this repository:
https://github.com/stat-ml/hist-loss.git
cd histloss
python setup.py install
pip install hist-loss
criterion = HistogramLoss()
positive = torch.sigmoid(torch.randn(10, requires_grad=True))
negative = torch.sigmoid(torch.randn(10, requires_grad=True))
loss = criterion(positive, negative)
loss.backward()