CVPR 2020: 8比特數(shù)值也能訓(xùn)練模型?商湯提出訓(xùn)練加速新算法
實(shí)驗(yàn)結(jié)果
圖像分類任務(wù):
本文在CIFAR10和ImageNet等圖像分類數(shù)據(jù)集進(jìn)行INT8訓(xùn)練實(shí)驗(yàn)。從下表結(jié)果中可以看出,在大多數(shù)網(wǎng)絡(luò)結(jié)構(gòu)中均取得了比現(xiàn)有最好方法更優(yōu)的精度,并且首次在MobileNet、Inception等網(wǎng)絡(luò)上進(jìn)行量化訓(xùn)練實(shí)驗(yàn),精度損失也在1.5%以內(nèi)。
目標(biāo)檢測任務(wù):
同時,本文也首次嘗試在PASCAL和COCO等目標(biāo)檢測數(shù)據(jù)集上進(jìn)行INT8訓(xùn)練實(shí)驗(yàn),精度損失也在2%以內(nèi)。
已有的少量探究梯度量化的論文[4]均未報告算法在實(shí)際訓(xùn)練任務(wù)中的真實(shí)加速性能,為了最大限度將方法實(shí)用化,本文在GeForce GTX1080TI顯卡上編寫并優(yōu)化了用于支持INT8訓(xùn)練的卷積前向和后向計算核心。實(shí)測結(jié)果表明,使用INT8卷積計算的前向和后向過程相比于浮點(diǎn)計算有明顯的加速,其中前向過程平均加速1.63倍,后向過程平均加速1.94倍。如下圖所示:
同時,本文在實(shí)際訓(xùn)練過程中進(jìn)行了完整的端到端測試,可以看到,INT8訓(xùn)練可以將ResNet50的一輪訓(xùn)練過程從0.360秒降低到0.293秒,整體訓(xùn)練過程提速了22%。
References
[1] Ruihao Gong, Xianglong Liu, Shenghu Jiang, TianxiangLi,Peng Hu, Jiazhen Lin, Fengwei Yu, and Junjie Yan. Differen-tiable softquantization: Bridging full-precisionand low-bitneural networks. In ICCV, October 2019.
[2] RundongLi, Yan Wang, Feng Liang, Hongwei Qin, Junjie Yan, and Rui Fan. Fully quantizednetwork for object detection. In The IEEE Conference on Computer Vision andPattern Recognition (CVPR), June 2019.
[3] Benoit Jacob,Skirmantas Kligys, Bo Chen, Menglong Zhu, Matthew Tang, Andrew Howard, HartwigAdam, and Dmitry Kalenichenko. Quantization and training of neural networks forefficient integer-arithmetic-only inference. 2018 IEEE Conference on ComputerVision and Pattern Recognition(CVPR), June 2018.
[4] Yukuan Yang, Shuang Wu, LeiDeng, Tianyi Yan, Yuan Xie, and Guoqi Li. Training high-performance andlarge-scale deep neural networks with full 8-bit integers, 2019.

發(fā)表評論
請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
即日-9.1立即下載>> 【限時下載】ADI中國三十周年感恩回饋助力企業(yè)升級!
-
即日-9.14點(diǎn)擊進(jìn)入 >> 【限時福利】TE 2025國際物聯(lián)網(wǎng)展·深圳站
-
10月23日立即報名>> Works With 開發(fā)者大會深圳站
-
11月27日立即報名>> 【工程師系列】汽車電子技術(shù)在線大會
-
精彩回顧立即查看>> 【在線研討會】解析安森美(onsemi)高精度與超低功耗CGM系統(tǒng)解決方案
-
精彩回顧立即查看>> 【在線會議】CAE優(yōu)化設(shè)計:醫(yī)療器械設(shè)計的應(yīng)用案例與方案解析
推薦專題
- 1 傳魏建軍與賈躍亭合作,長城汽車出海美國
- 2 黃仁勛:與雷軍長期合作,共探AI智駕
- 3 阿里首位程序員,“掃地僧”多隆已離職
- 4 DeepSeek R2加持,中國AI與芯片產(chǎn)業(yè)迎來新一輪協(xié)同進(jìn)化
- 5 先進(jìn)算力新選擇 | 2025華為算力場景發(fā)布會暨北京xPN伙伴大會成功舉辦
- 6 六大國產(chǎn)大模型,誰是最強(qiáng)“金融分析師”?|錦緞評測
- 7 2025年第一支10倍股,來了!
- 8 清華跑出具身智能獨(dú)角獸:給機(jī)器人安上眼睛和大腦,融資近20億
- 9 募資39.85億元!寒武紀(jì)押注大模型芯片與軟件平臺
- 10 國內(nèi)免費(fèi)版Deep Research上線,秘塔AI深度研究嘗試重塑知識工作范式