TorchJD:PyTorch多损失训练库

Skeylos2 · reddit · 2026-07-08

介绍PyTorch库TorchJD,用于多损失训练。对比标量化(将多个损失合成单一损失再做梯度下降)与雅可比下降(为每个损失分别计算梯度,再聚合成能同时降低各项损失的更新向量),后者在多目标严重冲突时更优。该库已实现文献中两类主要方法,并被正式收录进PyTorch生态。