Exploding and vanishing gradients in deep neural networks: the effect of residual connections
Vivek S Borkar
잔차 연결이 딥러닝의 기울기 폭발·소멸 문제를 완화하는 메커니즘을 에르고딕 이론으로 엄밀히 증명한 이론 연구.
심층 신경망 학습에서 기울기 폭발(exploding gradient)과 기울기 소멸(vanishing gradient)은 깊은 네트워크의 수렴을 어렵게 만드는 근본적인 문제다. 잔차 연결(residual connection)이 이러한 문제를 완화한다는 경험적 증거는 많았지만, 그 수학적 원리는 명확히 밝혀지지 않았다.
저자는 곱셈 에르고딕 이론(multiplicative ergodic theory)을 도입하여 신경망의 순전파를 선형화된 동역학계로 모델링한다. 특히 Furstenberg와 Kifer의 리아푸노프 지수(Lyapunov exponent) 특성화 정리를 활용하여, 잔차 연결이 리아푸노프 스펙트럼에 미치는 영향을 분석한다. 이를 통해 잔차 연결이 기울기 전파의 안정성을 어떻게 향상시키는지 수학적으로 증명한다.
본 연구는 잔차 연결이 기울기 폭발/소멸 문제를 완화하는 원리를 최초로 엄밀한 수학적 틀 안에서 설명했다. 이는 ResNet 등 잔차 연결을 사용하는 다양한 아키텍처의 성공에 대한 이론적 근거를 제공하며, 향후 더 깊고 안정적인 신경망 설계에 방향성을 제시한다.