핵심 개념
광학 신경망의 물리적 제약을 고려하여 양자화된 가중치로도 우수한 성능을 달성할 수 있는 양자화 인식 학습 프레임워크를 제안한다.
초록
이 연구에서는 광학 신경망의 물리적 제약을 고려하여 양자화된 가중치로도 우수한 성능을 달성할 수 있는 양자화 인식 학습 프레임워크인 QuATON을 제안한다.
광학 신경망은 디지털 마이크로미러 장치(DMD), 공간광변조기(SLM) 등의 광학 소자를 이용하여 구현되며, 각 소자의 상태가 학습 가능한 매개변수로 작용한다. 하지만 이러한 광학 소자는 물리적 제약으로 인해 연속적인 값을 가질 수 없고 양자화된 값만을 가질 수 있다.
이 문제를 해결하기 위해 저자들은 Progressive Sigmoid Quantization (PSQ)라는 새로운 양자화 함수와 점진적 학습 기법을 제안한다. PSQ 함수는 시그모이드 함수를 기반으로 하며, 온도 매개변수를 통해 점진적으로 하드 양자화 함수에 근접하도록 한다.
제안된 QuATON 프레임워크를 이용하여 회절 신경망(Diffractive Deep Neural Network, D2NN)을 학습한 결과, 기존 방법들에 비해 우수한 성능을 보였다. 특히 양자화 수준이 낮은 경우(2-4비트) 기존 방법 대비 50% 이상의 성능 향상을 달성했다.
이를 통해 저자들은 물리적 제약을 고려한 양자화 인식 학습 기법을 제안함으로써, 향후 고성능 광학 신경망 구현의 기반을 마련했다.
통계
양자화 수준이 낮을수록(2-4비트) 제안 방법의 성능 향상 폭이 크다.
MNIST 데이터셋에서 2비트 양자화 시 제안 방법의 분류 정확도가 75.03%로, 기존 방법 대비 50% 이상 향상되었다.
TinyImageNet 데이터셋에서 8비트 양자화 시 제안 방법의 SSIM이 0.6042로, 기존 방법 대비 약 10% 향상되었다.
인용구
"양자화된 가중치로도 우수한 성능을 달성할 수 있는 양자화 인식 학습 프레임워크를 제안한다."
"제안된 QuATON 프레임워크를 이용하여 회절 신경망(D2NN)을 학습한 결과, 기존 방법들에 비해 우수한 성능을 보였다."