刊名:计算机系统应用
主办:中国科学院软件研究所
ISSN:1003-3254
CN:11-2854/TP
语言:中文;
周期:月刊
影响因子:0.983
被引频次:84519
数据库收录:
JST 日本科学技术振兴机构数据库(日)(2022)
北京大学《中文核心期刊要目总览》来源期刊:
2000年版期刊分类:信息科技
期刊热词:
计算机软件及计算机应用
随着深度神经网络在计算机视觉领域的广泛应用,后训练量化技术成为实现神经网络在资源受限设备上高效部署的关键技术。然而,现有的混合精度后训练量化方法在低比特量化时,计算最优位宽的耗时较长,同时由于校准样本有限,导致测试精度下降,这限制了其在实际应用中的性能表现。针对这一问题,本文提出了基于Fisher信息引导的混合精度后训练量化算法。
本文提出的方法主要基于Fisher信息理论,通过分析神经网络中不同层和不同神经元的Fisher信息,来指导位宽的选择。具体而言,我们首先计算每个神经元的Fisher信息,然后根据这些信息确定每个神经元的位宽。在混合精度训练过程中,我们引入了自适应的学习率调整机制,以优化训练过程中的位宽选择。此外,我们还提出了一种基于迁移学习的策略,以解决校准样本有限的问题,通过在多个数据集上预训练,提高算法的泛化能力。
通过在多个基准数据集上的实验,验证了所提出算法的有效性。实验结果表明,与传统的混合精度后训练量化方法相比,本文提出的算法在低比特量化时能够显著减少计算耗时,同时保持了较高的测试精度。具体来说,我们的算法在4位量化时,相比基线方法,测试精度提升了1.5%,而计算时间减少了40%。此外,在有限的校准样本情况下,我们的算法也能有效提高测试精度。
本文提出的基于Fisher信息引导的混合精度后训练量化算法,通过优化位宽选择和引入迁移学习策略,有效解决了现有混合精度后训练量化方法中存在的问题。该方法不仅能够提高低比特量化时的计算效率,还能在样本有限的情况下保持较高的测试精度。这对于推动深度神经网络在资源受限设备上的应用具有重要意义,有望促进计算机视觉技术在更多领域的实际应用。
文章来源:《计算机系统应用》 网址: http://www.jsjxtyyzz.cn/qikandaodu/2026/1027/328.html