![Single-IMC and dual-IMC. Credit: Device (2024). DOI: 10.1016/j.device.2024.100546 New computing scheme could enhance machine learning, facilitate breakthroughs in AI](https://scx1.b-cdn.net/csz/news/800a/2024/new-computing-scheme-c-4.jpg)
像 ChatGPT 這樣的人工智慧 (AI) 模型運行在演算法上,對數據有很大的興趣,它們透過機器學習來處理數據,但是它們的數據處理能力的限制又如何呢?由北京大學積體電路學院和人工智慧研究院孫忠教授領導的研究人員著手解決限制資料處理的馮諾依曼瓶頸。
在他們的論文中發表在日記中裝置2024年9月12日,團隊研發出雙IMC(記憶體運算)方案,不僅加速了機器學習過程,同時也提高了傳統數據操作的能源效率。
在策劃演算法時,軟體工程師計算機科學家依賴稱為矩陣向量乘法 (MVM) 的數據運算,它支持神經網路。神經網路是一種模擬人腦功能和結構的人工智慧模型中常見的運算架構。
隨著資料集規模的快速成長,計算效能通常受到資料移動以及處理和傳輸資料之間速度不匹配的限制。這被稱為馮諾依曼瓶頸。傳統的解決方案是單記憶體計算(single-IMC)方案,其中神經網路權重儲存在記憶體晶片中,而輸入(例如影像)由外部提供。
然而,單一 IMC 需要注意的是片內和片外資料傳輸之間的切換,以及數位類比轉換器 (DAC) 的使用,這會導致電路佔用空間較大和功耗較高。
![Dual in-memory computing enables fully in-memory MVM operations. Credit: Device (2024). DOI: 10.1016/j.device.2024.100546 New computing scheme could enhance machine learning, facilitate breakthroughs in AI](https://scx1.b-cdn.net/csz/news/800a/2024/new-computing-scheme-c-3.jpg)
為了充分發揮 IMC 原理的潛力,團隊開發了一種雙 IMC 方案,將神經網路的權重和輸入儲存在記憶體陣列中,從而以完全記憶體中的方式執行資料操作。
然後,團隊在電阻式隨機存取記憶體 (RRAM) 裝置上測試了雙 IMC,以進行訊號恢復和影像處理。以下是雙 IMC 方案應用於 MVM 操作時的一些好處:
- 完全記憶體運算實現了更高的效率,從而節省了片外動態隨機存取記憶體 (DRAM) 和片上靜態隨機存取記憶體 (SRAM) 帶來的時間和能源
- 透過完全記憶體方式消除了限制因素資料移動,從而優化了計算效能。
- 由於消除了單 IMC 方案中所需的 DAC,因此生產成本更低。這也意味著節省晶片面積、運算延遲和功耗要求。
隨著需求的快速成長數據-在當今數位時代的處理中,這項研究的發現可能會帶來運算架構和人工智慧的新突破。
更多資訊:Shiqing Wang 等人,用於加速神經網路的矩陣向量乘法的雙記憶體計算,裝置(2024)。DOI:10.1016/j.device.2024.100546
引文:計算方案加速機器學習,同時提高傳統資料運算的能源效率(2024年9月26日)檢索日期:2024 年 9 月 28 日來自 https://techxplore.com/news/2024-09-scheme-machine-energy-efficiency-traditional.html
本文檔受版權保護。除了出於私人學習或研究目的的任何公平交易外,不得未經書面許可,不得複製部分內容。所提供的內容僅供參考。