跳到正文
原文
电子科学与工程学院·科技动态·· 2020-03-03AI 评分22

国家ASIC中心单伟伟、杨军团队在ISSCC 2020发表AI芯片亮点论文

AI 导读

东南大学国家ASIC中心单伟伟、杨军、时龙兴教授的团队在ISSCC 2020发表语音关键词唤醒低功耗AI加速器亮点论文。芯片从算法、架构和电路三层优化,在28nm CMOS下实现0.41V工作电压和510nW功耗。较并行FFT,串行FFT电路存储量降低8×、功耗降低11×;逐帧数据复用使计算量降低17.4×。

正文

2020 年 2 月 16-20 日,第 67 届 IEEE 国际固态电路峰会(ISSCC 2020)于美国旧金山

召开。这是世界上规模最大、最权威、水平最高的固态电路国际会议,被称为集成电路行业

的芯片奥林匹克大会。此次 ISSCC 的主题是“Integrated Circuits Powering the AI ERA”,

彰显了人工智能时代集成电路的重要地位。

本次会议新设了两个机器学习分会,且所有的四个主题演讲都与人工智能有关。两个机

器学习分会分别以高性能和低功耗为主题。其中“Low-Power Machine Learning”分会

(Session 14)面向的是可移动终端等对功耗有极致要求的领域,共收录 3 篇论文,而且都来

自于中国大陆,这展示了中国在 AI 芯片设计的学术领域已处于世界领先水平。亮点论文来

自东南大学国家 ASIC 中心单伟伟、杨军、时龙兴教授的团队。另两篇论文来自清华大学刘

勇攀教授团队和湃方科技,一篇是基于帧间数据计算复用的视频应用 AI 处理器,另一篇是

采用自适应稀疏系统级技术的存内计算神经网络处理器。

图 1 低功耗机器学习分会的 3 篇论文

我校的亮点论文是关于语音关键词唤醒的低功耗 AI 加速器的论文:"A 510nW, 0.41V

low-memory, low-computation keyword spotting chip using serial FFT based MFCC

and binarized depthwise separable convolutional neural network in 28nm CMOS"。

该语音唤醒智能芯片从算法、芯片架构和电路三个层次统筹优化,如下图所示,算法级采用

基于串行 FFT 的 MFCC 特征提取和深度可分离卷积神经网络,极大降低了计算量和存储量;

架构级提出了语音数据的逐帧数据复用方法。这两个级别联合使得芯片可工作在极低频率

40kHz 下,进而促成了全芯片的近阈值设计和超低漏电的定制存储器电路,最终现了史上功

耗最低的关键词唤醒电路,功耗仅为 510 纳瓦。

图 2 语音唤醒智能芯片的算法、芯片架构和电路三个层次统筹优化

独特的创新技术如下:

(1)提出并实现了基于串行 FFT 的梅尔频率倒谱系数(MFCC)特征提取电路,同时用

混合量化逐层降低硬件实现代价。FFT 是特征提取中计算最复杂、功耗最大的模块,与传统

并行 FFT 相比,提出的串行 FFT 电路的存储量降低 8×,功耗降低 11×;

图 3 MFCC 特征提取电路结构及亮点

(2)深度可分离卷积神经网络的二值化轻量级神经网络,与 CNN 相比存储量和计算量

均降低 7×;基于此设计了契合算法的神经网络硬件架构,由计算单元(PE)阵列(含 32 个

乘累加 MAC 单元)、存储模块、数据映射模块及控制状态机组成;

(3)提出了逐帧数据复用技术,利用语音应用中相邻两帧输入数据的计算存在大量重

复导致的卷积计算中存在大量重复计算的特点,对神经网络中的数据的存储和计算量进行压

缩,使得计算量降低 17.4×,中间存储量减少 3.5×;

10

FFT

Mel

filters

m1 m2 m3 m4 m5 m6 m9 m10m7 m8

10

Collected Speech signal

FFT

Stage 1

Data

Stage 2 ...

BF BF BF...

... 2N

Buffer

2N

Output

Parallel FFT

2N

Input

2N

Buffer

...

VS

... Out

Data

BF BF

Buffer2N-1 Buffer2N-2

W

BF BF

Buffer2 Buffer1 W

In Data Serial FFT

Parallel FFT

Serial FFT

FFT comparison Storage

2N

NN 2.

MFCC Features

MFCC

circuit

time/ms

frame1

frame2

frame3

frame29

frame30

frame31

V

16ms

32ms

21

22

.

.

.29

30

31

updated

1

updated17

18

19

20

32

10 4 updated2

32

updated

2

32

321

9

updated

OR

1

3

Weight

CONV DSC DSCPW POOLING FC

Gray box:

eliminated

storage

1bit 1bit 1bit

1bit

0 1 2 8 9···

10 dim / frame

kw1

kw2

filter

1

2

.

.

.

19

20

1

2

.

.

.

1

2

.

.

.

17

18

1

2

.

.

.

17

16

18

Odd

Even

4

10

32

32

1

9

updated

1bit

33

32

1

1

32 32

3

288

Input features

10 dimension

*8b/frame

ACT1 ACT2 ACT3

Speech data MFCC features

Black box:

storage

Conv kernel

10*4*

32 channels

Only 1 frame is

different

DSC NN hardware with elimination of redundant storage and computations

Original Proposed

Conv

MEM Calculation MEM/bit Calculation

3× 20× 0.8K 5.12K

DSC 6.7× 18× 0.384K 0.576K

DSCPW 18× 18× 0.064K 2.05K

Pooling 9× 9× 0.128K 0.128K

FC 0.5× 1× 0.576K 0.86K

Memory (MEM) and Calculation Reduction Effect

图 4 深度可分离卷积神经网络架构及逐帧数据复用技术

(4)全芯片近阈值设计与定制的极低漏电存储器,再次降低功耗。前面算法级和架构

级的双重优化,使得整体神经网络加速器仅需 640 个周期就能完成一轮推理,在 16ms 的帧

间隔内完成即可,因此工作频率仅需 40kHz 下,这就促成了全芯片可采用近阈值设计。近阈

值设计的难点有二:存储器和漏电控制。我们定制了能工作在低电压下、且具有低漏电的

latch 型存储器,实现神经网络与 MFCC 所需的片上多块、多类型的小容量存储。定制 Memory

比工艺厂提供的 SRAM 编译器生成的同等大小的 SRAM 模块的漏电低了 12 倍,且可与其他数

字电路一起工作在近阈值 0.41V。

这些技术应用在 T28nm 的神经网络加速器,实现了史上最低的关键词唤醒电路芯

片,0.41V 电压下整体功耗仅 0.51uW,比国际同类研究降低了 10 到 564 倍,见对比表格。

ISSCC2017[2] VLSI2018[3] VLSI2019[4] ESSCIRC18[5]

Tech. 40 nm 28 nm 65 nm 65 nm

Voltage 0.63-0.9 V 0.57-0.9 V 0.6 V 0.575 V

Frequency 1.9 MHz 2.5 MHz 250 KHz 250 KHz

Latency 6.5 ms 0.5-25 ms 16 ms 16 ms

Keyword Num 10 words 1 word 10 words 4 words

Power 288 μW 141 μW 16.1 μW* 5 μW**

This work

28 nm

0.41 V

40 KHz

16 ms

2 words

0.51 μW

* 16.1 μW refers to the power of digital KWS system in [4], not including Analog Front-End.

* * Ref. [5] does not have MFCC circuit.

Dataset NA TIDIGIT GSCD NA GSCD

Accuracy NA 96% 90.87% 91.2% 93.6%

Memory 270 KB 52 KB 65 KB 32 KB

Core Size 7.1 mm2 1.29 mm2 2.56 mm2 1.04 mm2

2 KB

0.23 mm2

Algorithm DNN CNN LSTM LSTM DSCNN

来源:电子科学与工程学院·科技动态 · electronic.seu.edu.cn