PyTorch 多分类损失函数

PolyLoss用1行代码+1个超参完成超车

ICLR 2022 超越Focal Loss PolyLoss用1行代码+1个超参完成超车_AI视觉网奇的博客-CSDN博客

将“softmax+交叉熵”推广到多标签分类问题

将“softmax+交叉熵”推广到多标签分类问题 - 科学空间|Scientific Spaces

多标签分类问题：

torch版：

import torch
import numpy as np
from scipy.special import logsumexp
import torch.nn.functional as F
def multilabel_categorical_crossentropy(y_true, y_pred):
    y_pred = (1 - 2 * y_true) * y_pred
    y_pred_neg = y_pred - y_true * 1e12
    y_pred_pos = y_pred - (1 - y_true) * 1e12
    zeros = torch.zeros_like(y_pre