How to use temperature scaling for binary classification using sigmoid?

[evapachetti]

My network is trained according to a binary classification approach, so the model outputs as a single logit value which I then convert into probability by applying the sigmoid function. How can I modify the temperature scaling code to apply it to my network?

Thank you in advance.

[chenxia-han]

I think the following approach should work:

1. Replace the CrossEntropyLoss with BCEWithLogitsLoss
2. Change the size of self.temperature to number of classes

Sigmoid can be regarded as a special case of softmax where one of the logits is 0: $\frac{e^x}{e^0 + e^x}$. Then we only need to learn the temperature for each class.