본 문서는 종단간(End-to-End) 음성 인식의 역사적 이정표인 Baidu Research의 Deep Speech 2(arXiv:1512.02595, ICML 2016)를 분석한다. 기존 ASR 파이프라인의 수작업 컴포넌트를 CTC 기반 심층 순환신경망으로 대체하고, HPC 수준의 분산학습 인프라로 영어·중국어 두 언어에서 인간 전사자 수준의 정확도를 달성한 연구이다.
기존 ASR 파이프라인은 특징 추출, 음향 모델, 언어·발음 모델, 화자 적응 등 수십 년간 축적된 다수의 독립 컴포넌트로 구성되어, 새로운 언어나 환경으로의 이식이 매우 어렵다. Deep Speech 2(DS2)는 아키텍처, 대규모 라벨 데이터, 계산 규모라는 세 축을 동시에 개선하여 이 문제를 해결한다. HPC 기법 적용으로 전작(DS1) 대비 7배의 학습 속도 향상을 달성하였고, 이를 통해 영어 오류율을 최대 43% 감소시켰으며 중국어(Mandarin)도 높은 정확도로 인식하였다.
1. 서론: 종단간 학습으로의 전환
저자들은 "모델이 학습해야 할 지식"과 "그 지식을 얻기 위한 수작업 파이프라인"을 분리하고자, 아키텍처·데이터·계산 규모를 동시에 확장하는 접근을 취한다. DS2는 영어 11,940시간, 중국어 9,400시간의 라벨 데이터로 학습되었으며, 여러 벤치마크에서 Amazon Mechanical Turk 인간 작업자의 전사 성능과 경쟁하거나 이를 능가하였다.
2. 관련 연구
종단간 ASR 접근은 크게 두 갈래로 나뉜다. 첫째는 RNN 인코더-디코더(+어텐션) 구조로, 가변 길이 입력을 고정 길이 벡터로 인코딩한 뒤 디코딩한다. 둘째는 본 논문이 채택한 CTC(Connectionist Temporal Classification) + RNN 구조로, 프레임 단위 정렬 없이 처음부터(from scratch) 학습이 가능하다는 장점이 있다. 딥러닝에서의 스케일 활용(대규모 데이터와 분산학습)은 이미 컴퓨터 비전 등에서 성공의 핵심 동력으로 확인된 바 있다.
3. 모델 아키텍처
3-1. 기본 구조
DS2는 하나 이상의 컨볼루션 레이어, 하나 이상의 순환(RNN/GRU) 레이어, 하나 이상의 완전연결 레이어, 그리고 CTC 출력층으로 구성된다. 출력은 문자 단위 확률분포이며, 클리핑된 ReLU \(\sigma(x)=\min(\max(x,0),20)\)를 비선형함수로 사용한다.
3-2. Batch Normalization for Deep RNNs
모델 용량을 깊이로 확장하기 위해, 순환신경망에 배치 정규화를 적용하는 방법을 탐구하였다.
$$B(x) = \gamma\frac{x - E[x]}{\sqrt{\text{Var}[x] + \epsilon}} + \beta$$
매 time-step마다 정규화를 적용하는 방식은 시간축 순차 의존성으로 인해 효과가 없었으며, 대신 미니배치 내 시퀀스 전체 길이에 걸쳐 평균·분산을 계산하는 시퀀스 단위(sequence-wise) 정규화가 효과적이었다. 38M 파라미터로 고정한 실험에서, 시퀀스 단위 배치 정규화는 네트워크가 깊을수록 개선폭이 커져 가장 깊은 모델(7개 순환층, 총 9층)에서 12%p의 성능 차이를 보였다. 다만 얕은 모델에서는 오히려 일반화 성능을 해쳤다. 배포 시에는 미니배치 통계 대신 학습 중 수집한 이동평균을 사용하여 단일 발화에 대해서도 정확한 평가가 가능하도록 하였다.
3-3. SortaGrad
CTC 손실 함수는 발화 길이가 길수록 정렬 경로의 곱셈 항이 많아져, 암묵적으로 긴 발화에 더 큰 손실 스케일을 부여한다.
$$\mathcal{L}_{ctc}(x,y;\theta) = -\log \sum_{\ell \in \text{Align}(x,y)} \prod_t p_{ctc}(\ell_t | x;\theta)$$
이를 커리큘럼 학습 신호로 활용한 것이 SortaGrad이다. 첫 번째 epoch는 발화 길이(난이도의 프록시)의 오름차순으로 미니배치를 순회하고, 이후 epoch부터는 무작위 순서로 복귀한다. 이 기법은 배치 정규화가 적용되지 않은 모델에서 특히 효과가 두드러졌으며, 수치적으로 불안정한 초월함수 구현 차이(CPU와 GPU 간)로 인한 문제도 완화하였다.
3-4. Simple RNN과 GRU 비교
GRU는 다음과 같이 정의된다.
$$z_t = \sigma(W_z x_t + U_z h_{t-1} + b_z), \quad r_t = \sigma(W_r x_t + U_r h_{t-1} + b_r)$$ $$\tilde{h}_t = f(W_h x_t + r_t \circ U_h h_{t-1} + b_h), \quad h_t = (1-z_t)h_{t-1} + z_t \tilde{h}_t$$
동일 파라미터 규모(38M)에서는 GRU가 Simple RNN보다 일관되게 낮은 WER을 보였다. 그러나 이후 100M 파라미터 규모로 확장했을 때는 오히려 Simple RNN이 GRU를 근소하게 앞섰고 학습 속도도 더 빨라, 최종 대규모 실험에서는 Simple RNN을 채택하였다.
3-5. 주파수 컨볼루션
시간 축만을 다루는 1D 컨볼루션과 시간·주파수를 함께 다루는 2D 컨볼루션을 비교한 결과, 2D 컨볼루션은 특히 잡음이 있는 데이터셋에서 큰 개선을 보였다(3-layer 2D가 3-layer 1D 대비 잡음 데이터셋에서 23.9%의 상대적 WER 개선). 주파수 방향의 컨볼루션이 화자 변이에 따른 스펙트럼 분산을 완전연결·순환 레이어보다 효율적으로 모델링하는 것으로 해석된다.
3-6. Striding과 바이그램 출력
컨볼루션에 긴 stride를 적용하면 연산량은 줄어들지만, 영어는 출력 문자 수 대비 필요한 최소 시간 스텝 제약으로 인해 정확도가 저하된다. 저자들은 영어와 중국어의 정보 밀도 차이를 정량적으로 분석하였다: 영어는 평균 14.1문자/초(4.9bit/문자), 중국어는 3.3문자/초(12.6bit/문자)로, 시간당 엔트로피 밀도는 영어 약 41bit/초, 중국어 약 58bit/초이다. 이 차이를 극복하기 위해 영어에 비중첩 바이그램(non-overlapping bigram) 출력을 도입하였으며, 이를 통해 유니그램보다 훨씬 큰 stride에서도 WER 손실 없이 연산·메모리 이득을 얻을 수 있었다.
3-7. Row Convolution과 단방향 모델
양방향 RNN은 전체 발화를 필요로 하여 저지연 온라인 배포에 부적합하다. 저자들은 모든 순환 레이어 위에 하나의 Row Convolution 레이어를 추가하여, 순수 단방향(forward-only) 구조로도 양방향 모델에 준하는 성능을 달성하였다.
$$r_{t,i} = \sum_{j=1}^{\tau+1} W_{i,j}\, h_{t+j-1,i}, \quad 1 \le i \le d$$
이는 현재 시점에서 작은 미래 문맥(\(\tau\)스텝)만 참조해도 충분하다는 직관에 기반한다. Row Convolution을 순환 레이어 위에 배치함으로써 그 아래 연산을 세밀한 단위로 스트리밍할 수 있었고, 흥미롭게도 중국어에서는 최고 성능의 양방향 모델보다 더 나은 CER을 달성하였다.
3-8. 언어모델 통합
수백만 발화로 학습된 RNN 자체가 강력한 암묵적 언어모델 능력을 갖추지만, 라벨 데이터보다 압도적으로 큰 비표기 텍스트 코퍼스를 활용하기 위해 외부 n-gram 언어모델을 결합하였다. 영어는 Common Crawl 텍스트로 학습한 Kneser-Ney 평활 5-gram(약 8.5억 n-gram), 중국어는 내부 텍스트 80억 줄로 학습한 문자 단위 5-gram(약 20억 n-gram)을 사용한다. 디코딩 시에는 다음 점수를 최대화하는 빔서치를 수행한다.
$$Q(y) = \log(p_{ctc}(y|x)) + \alpha \log(p_{lm}(y)) + \beta \cdot \text{word\_count}(y)$$
네트워크가 깊어질수록(5층에서 9층으로) 언어모델의 상대적 기여도는 영어 48%→36%, 중국어 27%→23%로 감소하였는데, 이는 더 깊은 네트워크일수록 강력한 암묵적 언어모델을 스스로 학습함을 시사한다.
4. 시스템 최적화(HPC)
모델 하나의 학습에는 수십 exaFLOPs가 필요하여 단일 GPU로는 3~6주가 소요된다. 저자들은 C++ 딥러닝 라이브러리와 CUDA 기반 선형대수 라이브러리를 자체 구축하여, 노드당 8개의 Titan X GPU에서 이론 최대치의 45%인 24 teraFLOP/s를 달성하였다.
- 동기 SGD와 데이터 병렬화: 비동기 파라미터 서버 방식 대신 동기 SGD를 채택하여 재현성과 디버깅 용이성을 확보하였다.
- 커스텀 all-reduce: 링 알고리즘을 직접 구현하여 OpenMPI 대비 8~16GPU 학습 전체 기준 2.5배의 학습 가속을 달성하였다.
- GPU 기반 CTC 손실 계산: 기존 CPU 구현의 확장성 병목을 해소하여, 영어 기준 28.9배, 중국어 기준 12.5배의 CTC 계산 가속을 달성하였다.
- 커스텀 메모리 할당기: 버디 알고리즘 기반의 자체 할당기를 구현하여, 표준 malloc 대비 최대 2배 이상의 오버헤드를 제거하였다.
5. 학습 데이터
영어 11,940시간(WSJ, Switchboard, Fisher, LibriSpeech, Baidu 자체 코퍼스), 중국어 9,400시간의 데이터로 학습되었다. 긴 원본 오디오는 기존 CTC-RNN으로 Viterbi 정렬을 수행하고, 연속된 blank 구간을 기준으로 분할한 뒤, 정렬 실패 예시를 선형 분류기로 걸러내는 3단계 파이프라인을 통해 정제하였다. 이 필터링으로 영어 데이터의 WER이 17%에서 5%로 개선되면서도 예시의 50% 이상이 보존되었다. 데이터 증강으로는 무작위로 선택한 발화의 40%에 노이즈를 추가하였다.
데이터 규모를 1%(120시간)에서 100%(12,000시간)까지 늘렸을 때 WER은 거듭제곱 법칙(power law)을 따라 감소하였으며, 데이터가 10배 증가할 때마다 WER은 상대적으로 약 40% 감소하였다.
6. 실험 결과
6-1. 영어
모델 크기를 18M에서 100M 파라미터로 확장하는 실험에서, 성능은 100M 파라미터까지 일관되게 개선되었다. DS2(100M, Simple RNN)는 DS1(5층, 1개 순환층) 대비 Baidu 내부 테스트셋에서 24.01%에서 13.59%로 43.4%의 상대적 개선을 보였다.
| 테스트셋 유형 | DS1 | DS2 | 인간 |
|---|---|---|---|
| WSJ eval'92 (낭독체) | 4.94 | 3.60 | 5.03 |
| LibriSpeech test-clean | 7.89 | 5.33 | 5.83 |
| VoxForge Indian (억양) | 45.35 | 22.44 | 22.15 |
| CHiME eval real (실제 잡음) | 67.94 | 21.79 | 11.84 |
낭독체 음성에서는 4개 테스트셋 중 3개에서 인간을 능가하였다. 반면 실제 잡음 환경(CHiME eval real)에서는 DS1 대비 크게 개선되었음에도 여전히 인간 수준에 미치지 못했으며, 특히 실제 잡음이 합성 잡음보다 인간과의 격차가 더 컸다.
6-2. 중국어
얕은 RNN 대비 9층·7개 순환층·배치 정규화·2D 컨볼루션을 갖춘 모델이 48%의 상대적 개선(dev 7.13%→5.81%)을 보여, 영어와 동일하게 "더 깊은 양방향 순환"이 핵심 개선 요인임이 재확인되었다. 무작위 100개 발화에 대해 5인 그룹 전사와 비교했을 때 인간 오류율 4.0% 대비 시스템 3.7%로 시스템이 근소 우위였으며, 250개 발화에 대해 단일 전사자와 비교했을 때는 인간 9.7% 대비 시스템 5.7%로 시스템이 크게 우수하였다.
7. 배포
연구용 모델(양방향, 전체 발화 필요, 넓은 빔서치)은 실시간 배포에 부적합하다. 저자들은 중국어 시스템을 중심으로 다음의 배포 최적화를 설계하였다.
- Batch Dispatch: 사용자 요청 스트림을 배치로 묶어 GPU 연산 효율을 높이는 스케줄러. 배치가 완료되는 즉시 다음 배치를 처리하는 eager 배칭 방식이 지연시간 최소화에 가장 효과적이었으며, 10개 동시 스트림 기준 중앙값 44ms, 98퍼센타일 70ms의 지연시간을 달성하였다.
- 16비트 반정밀도 행렬곱 커널: 배포 시 정확도 저하 없이 메모리 대역폭을 절약하는 자체 커널을 구현하여, Nervana Systems 및 NVIDIA CUBLAS 대비 우수한 성능을 보였다.
- 빔서치 가지치기: 누적확률 임계값(p=0.99)과 최대 문자 수 제한을 통해 언어모델 조회를 150배 가속하면서도 CER에는 미미한 영향(0.1~0.3% 상대)만 미쳤다.
최종적으로 연구용 모델의 CER 5.81%는 배포 모델에서 6.10%로, 단 5%의 상대적 성능 열화만으로 저지연 배포를 달성하였다. 배포 모델은 순방향 전용 순환 5층(2560 유닛), Row Convolution 1층(\(\tau=19\)), 완전연결 1층으로 구성된다.
8. 결론
종단간 딥러닝은 데이터와 연산의 증가에 비례하여 지속적으로 성능이 개선되는 실용적 경로임이 실증되었다. 서로 다른 두 언어(영어·중국어)에 언어 전문 지식 없이 동일한 방법론을 적용할 수 있음을 보였으며, GPU 서버에서의 배칭을 통해 이를 효율적으로 배포할 수 있음도 함께 실증하였다.
9. 연구사적 의의
본 논문은 CTC 기반 종단간 ASR을 산업 규모(HPC 분산학습)로 처음 실증한 대표적 사례로, 이후 등장한 Whisper·Conformer·RNN-T 계열 연구 전반의 스케일링 철학(더 큰 모델·데이터·연산이 곧 성능 향상)의 직접적 선구자다. SortaGrad와 시퀀스 단위 배치 정규화는 지금도 시퀀스 모델 학습 안정화의 기본기로 유효하며, Row Convolution은 최근 스트리밍 ASR 연구에서 사용되는 lookahead 기반 아키텍처의 초기 형태로 볼 수 있다. 다만 순수 CTC 기반 접근은 외부 언어모델을 반드시 필요로 한다는 한계를 명시적으로 드러내는데, 이는 이후 연구들이 언어모델 지식을 인코더·디코더에 내재화하려는(LALM 등) 방향으로 나아간 배경을 이해하는 좋은 대조점이 된다.
부록: 전문용어 정리
- CTC (Connectionist Temporal Classification): 프레임별 예측과 blank 토큰으로 가변 길이 정렬 문제를 학습 가능하게 만드는 손실 함수.
- SortaGrad: 발화 길이를 난이도 프록시로 삼아 첫 epoch만 오름차순으로 학습시키는 커리큘럼 기법.
- 시퀀스 단위 BatchNorm: 미니배치 내 전체 시퀀스 길이에 걸쳐 평균·분산을 계산해 순환망에 적용하는 정규화 기법.
- GRU (Gated Recurrent Unit): 업데이트·리셋 게이트로 장기 의존성을 모델링하는 경량 순환 유닛.
- Row Convolution: 순환 레이어 위에서 작은 미래 문맥만 참조하는 레이어. 단방향 스트리밍 모델을 가능하게 함.
- 비중첩 바이그램 출력: 영어 알파벳을 2글자 단위로 묶어 출력 시퀀스 길이를 줄이는 기법.
- Batch Dispatch: 여러 사용자 요청을 실시간으로 모아 배치로 묶는 배포용 스케줄러.
- All-reduce (Ring algorithm): 다중 GPU 간 그래디언트를 교환·합산하는 분산학습 통신 패턴.
- 동기 SGD: 모든 워커가 매 스텝 그래디언트를 동기화 후 갱신하는 분산 학습 방식.
- WER / CER: 단어 및 문자 단위 오류율.
- Kneser-Ney 평활 n-gram: 통계적 언어모델의 대표적 평활 기법.
참고문헌
- Amodei, D. et al. (2016). Deep Speech 2: End-to-End Speech Recognition in English and Mandarin. Proceedings of ICML. arXiv:1512.02595.
- Hannun, A. et al. (2014). Deep Speech: Scaling up end-to-end speech recognition. arXiv:1412.5567.
- Graves, A., Fernández, S., Gomez, F., & Schmidhuber, J. (2006). Connectionist temporal classification: Labelling unsegmented sequence data with recurrent neural networks. Proceedings of ICML, 369–376.
- Ioffe, S., & Szegedy, C. (2015). Batch normalization: Accelerating deep network training by reducing internal covariate shift. arXiv:1502.03167.
※ 본 문서는 Deep Speech 2 논문(arXiv:1512.02595v1, 2015-12-08; ICML 2016)의 내용을 요약·정리한 것이다.
