Large language model accelerator architecture based on three-dimensional NAND flash memory

By using an accelerator architecture based on 3D NAND flash memory and a dynamic scheduling method, the storage and computation bottlenecks of large language models are solved, efficient matrix-vector multiplication and addition operations are achieved, inference performance and energy efficiency are improved, and hardware costs are reduced.

CN121501740AActive Publication Date: 2026-02-10SOUTHEAST UNIV
3 Cites 5 Cited by

Patent Information

Application Number
CN202511558183.3
Authority / Receiving Office
CN · China
Patent Type
Applications(China)
Current Assignee / Owner
Filing Date
2025-10-29
Publication Date
2026-02-10
Estimated Expiration
2045-10-29

AI Technical Summary

Technical Problem

There are bottlenecks in the computation and storage of large language models in existing technologies. In particular, in DRAM-based architectures, it is difficult to balance storage capacity and access bandwidth requirements. Weight read latency is high, data migration overhead is large, and the lack of dynamic scheduling mechanism leads to resource waste.

Method used

The design incorporates a large language model accelerator architecture based on 3D NAND flash memory, including DRAM, auxiliary computing units, NAND array chips, flash logic chips, and interconnect channels. By performing matrix-vector multiplication and addition operations within the flash memory and combining dynamic scheduling methods, the attention mechanism and feedforward neural network computation are optimized.

Benefits of technology

It improves the inference performance and energy efficiency of large language models, reduces weight transfer overhead, alleviates the memory wall bottleneck, supports pipelined access and dynamic scheduling, and reduces hardware costs.

✦ Generated by Eureka AI based on patent content.
Patent Text Reader

Abstract

The invention discloses a large language model accelerator architecture based on a three-dimensional NAND flash memory, and belongs to the technical field of calculation, reckoning or counting. The architecture comprises a three-dimensional NAND flash memory used for executing feedforward neural network calculation; the auxiliary calculation unit is used for executing attention mechanism calculation; the DRAM chip is used for storing attention mechanism related weights and KV cache; and the interconnection resource is used for realizing data interaction among the components. Wherein the three-dimensional NAND flash memory comprises a logic chip and an NAND array chip, the logic chip is used for controlling and executing calculation, and the NAND array chip is used for storing weights and participating in calculation. The invention further provides a scheduling method based on KV cache awareness. The scheduling method comprises decomposition and dynamic allocation of calculation tasks. Through collaborative design of the hardware architecture and the scheduling method, the memory wall bottleneck in large language model calculation can be relieved, the energy consumption is reduced, and the overall calculation performance is improved.
Need to check novelty before this filing date? Find Prior Art