多目标凸量化模型

高效模型压缩

将网络训练建模为多目标优化,同时追求高精度和低量化误差,通过可微凸误差函数和动态拉格朗日系数自适应平衡两个目标。

樊春晓 · 郭丹 · Ziqi, Wang · Meng, Wang

IEEE Transactions on Pattern Analysis and Machine Intelligence 2024

技术优势

端到端可微

设计了可微的量化误差函数,避免量化操作不可导的反向传播问题,使量化可以与训练一起优化。

训练更稳定

采用时间序列自蒸馏训练方案,结合历史软标签与硬目标,保证训练过程中性能收敛可控且稳定。

自动平衡损失

动态拉格朗日系数自适应调整量化损失和性能损失的梯度幅度,无需手动调参即可平衡两个目标。

多基准验证

在MNIST、CIFAR-10/100、ImageNet、Penn Treebank和Microsoft COCO上评估,取得优于现有方法的结果。

应用场景