기술 요약: 완전 다중화된 광학 텐서 컴퓨팅 (FieldCore)

문제 정의

텐서 연산, 특히 행렬 곱셈과 컨볼루션은 현대 인공지능 및 고처리량 신호 처리의 계산적 중추입니다. 전자 가속기는 프로세싱 유닛의 복제된 배열을 통해 확장되어 왔으나, 메모리 계층 구조와 상호 연결 간의 데이터 이동 비용과 관련된 점점 더 심각한 병목 현상에 직면해 있습니다. 광 컴퓨팅은 광학장의 고유한 병렬성을 활용함으로써 유망한 대안을 제시합니다. 그러나 기존의 광 프로세서는 일반적으로 고립된 차원(예: 파장 분할 다중화 또는 시간 영역 처리)만을 활용하거나 일부 차원만을 결합하여 사용합니다. 핵심적인 과제는, 더 많은 광학 차원이 관여함에 따라 비직교성, 분산, 제조 공정의 불완리성 등의 결함으로 인해 전체 신호 공간에 걸친 계산의 균일성이 깨진다는 점입니다. 결과적으로, 모든 가이드된 광 신호의 고유한 다중화 차원(파장, 무선 주파수(RF), 가이드 모드, 공간, 시간)을 동시에 체계적으로 활용하면서도 전체 신호 공간에 걸쳐 고충실도 및 균일한 계산을 유지할 수 있는 일반적인 프레임워크가 부족한 실정입니다.

방법론

저자들은 파장, 무선 주파수(RF) 부반송파, 가이드 모드, 공간, 시간이라는 다섯 가지 차원을 공동으로 활용하여 이러한 한계를 극복하도록 설계된 완전 다중화된 광학 텐서 코어인 FieldCore를 소개합니다.

아키텍처 및 원리

FieldCore는 가이드된 광장이 여러 직교 차원으로 확장되어 고차 텐서를 형성할 수 있다는 원리로 작동합니다. 시스템은 입력 데이터를 이러한 차원들에 매핑하여, 단일 광학 칩이 곱셈적 병렬성(multiplicative parallelism)을 수행할 수 있도록 합니다.

• 신호 표현: 광장은 광 캐리어 주파수($\omega_\ell$), 벡터 가이드 고유 모드($\psi_m$), 그리고 RF 부반송파 오프셋($\Omega_n$)의 합으로 표현됩니다.
• 하드웨어 구현: 코어는 실리콘 온 인슐레이터(SOI) 플랫폼 상의 $K \times R$ 크로스바로 구현됩니다. 컴팩트한 크기, 광대역 응답, 제조 허용 오차를 보장하기 위해 역설계(inverse-designed)된 빌딩 블록을 통합했습니다.
• 주요 구성 요소:
  • 역설계된 디지털 메타물질: 모드 다중화기/역다중화기(MUX/deMUX), 3-dB 스플리터, 도파로 교차로(waveguide crossings)에 사용됩니다. 이를 통해 마이크로미터($\mu$m) 규모의 풋프린트 내에서 저크로스토크 및 광대역 동작을 가능하게 합니다.
  • 모드 불감성 위상 변조기 (MIPS): TE0 및 TE1 모드의 열광학 응답을 균등화하기 위해 도파로 폭을 넓힘으로써 설계되었으며, 이를 통해 모드 차원에 대한 가중치 불변성을 보장합니다.
  • 프로그래밍 가능한 라우팅 및 가중치 부여: 튜너블 커플러(듀얼 포트 MZI)와 곱셈기(싱글 포트 MZI 감쇠기)를 통해 신호를 분배하고 아날로그 가중치를 적용합니다.
  • 누적 (Accumulation): 튜너블 애더(adders)는 공간 입력 채널 전반에 걸쳐 가중 합산을 수행합니다.

동작 흐름

1. 입력: 시간 스트림이 RF 부반송파로 업컨버전되고, 특정 파장에 변조되며, 온칩 MUX를 통해 가이드 모드로 매핑됩니다.
2. 처리: 신호는 크로스바로 라우팅되어 병렬 가중 누적 과정을 거칩니다. 동일한 가중치 행렬이 모든 다중화된 슬라이스(RF, 파장, 모드, 시간)에 걸쳐 균일하게 적용됩니다.
3. 출력: 다중화 과정이 광학 및 전기 영역에서 역으로 수행되어 병렬 텐서 슬라이스를 복구합니다.

주요 기여

1. 통합 다차원 텐서 코어: FieldCore는 파장, RF 부반송파, 가이드 모드, 공간, 시간을 단일 프레임워크 내에 통합하여, 고유한 병렬 차원을 곱셈적 텐서 수준의 병렬성으로 변환하는 최초의 통합 광학 텐서 코어입니다.
2. 차원 균일 계산: 이 설계는 모든 차원에서 고충실도 계산을 달성합니다. 이는 다음을 통해 검증되었습니다:
   • 모드 불변성: MIPS를 사용하여 TE0 및 TE1 모드에 대해 일관된 가중치를 유지합니다.
   • 광대역 평탄도: 파장 및 RF 차원에 대해 스펙트럼 평탄한 응답을 보장합니다.
3. 역설계된 광학 빌딩 블록: 역설계된 디지털 메타물질 구성 요소를 사용하여 컴팩트하고 광대역(>40 nm)이며 저크로스토크(<-20 dB)인 멀티모드 동작을 가능하게 하며, 이는 채널 일관성을 유지하는 데 필수적입니다.
4. 확장 가능한 아키텍처: 시스템은 단일 코어 내에서 최대 1,800개의 병렬 입력 스트림을 지원하여, 광학 텐서 컴퓨팅을 위한 확장 가능한 패러다임을 구축합니다.

실험 결과

저자들은 $4 \times 4$ FieldCore 칩(2.55 $\times$ 2.58 mm$^2$)을 제작하였으며, 산술 검증 및 복합 추론 작업을 통해 성능을 검증했습니다.

산술 검증

• 속도 및 정밀도: 시스템은 TE0 및 TE1 모드 모두에 대해 최대 120 GBaud의 심볼 레이트에서 5비트 이상의 정밀도를 가진 곱셈-누적(MAC) 연산을 시연했습니다.
• 파장 안정성: C-밴드(1530–1565 nm) 전반의 정밀도 편차는 1550 nm 기준 대비 0.7비트 미만이었습니다.
• RF 확장성: 시스템은 단일 광 캐리어에서 최대 100개의 RF 부반송파를 지원하며, 총 보드레이트 50 GBaud에서 약 5비트의 덧셈 정밀도를 유지했습니다.

태스크 수준 시연

• 이미지 컨볼루션:
  • 그레이스케일: 네 개의 2 $\times$ 2 커널을 사용하여 150 $\times$ 150 이미지를 대상으로 방향성 에지 추출을 수행하였으며, 10 GBaud에서 30.5 dB, 70 GBaud에서 23.4 dB의 피크 신호 대 잡음비(PSNR)를 달성했습니다.
  • RGB: FDM을 사용하여 RGB 이미지를 성공적으로 전송 및 컨볼루션하였으며, 채널과 모드 전반에 걸쳐 구조적 세부 사항을 보존했습니다.
• MNIST 필기 숫자 인식:
  • 파장, 모드, RF 다중화를 결합하여(200개 병렬 데이터 레인) 필기 숫자를 처리했습니다.
  • 모든 다중화 채널에서 평균 **93.8%**의 정확도를 달로 달성했으며, 이는 95.0%의 디지털 베이스라인과 비교됩니다.

고차원 및 병렬 애플리케이션

• 채널 병렬 초분광 분류: Indian Pines 데이터셋의 200개 스펙트럼 밴드를 200개의 병렬 채널에 매핑했습니다. 시스템은 12개 지표 피복 클래스에 대해 90.8%의 전체 분류 정확도를 달성했으며, 이는 5.16비트의 등가 아날로그 정밀도에 해당합니다.
• 인스턴스 병렬 기계 고장 진단: CWRU 베어링 데이터셋의 200개 동시 진동 신호 인스턴스를 처리했습니다. 시스템은 7가지 작동 조건(정상 및 6가지 고장 유형)에 대해 95.1%의 전체 정확도를 달성하였으며, 각 채널별로 견고한 성능을 보였습니다.

의의 및 주장

본 논문은 FieldCore가 고처리량 및 대규모 병렬 광학 텐서 프로세싱을 위한 확장 가능한 경로를 구축한다고 주장합니다. 가이드된 광 신호의 모든 고유한 다중화 차원을 활용함으로써, FieldCore는 현재의 전자 시스템 및 부분적으로 다중화된 광 시스템의 한계를 해결합니다.

• 처리량: 저자들은 처리량 극대화 모드에서 **69.12 Tera Operations Per Second (TOPS)**의 총 계산 처리량을 추정합니다.
• 병렬성: 아키텍처는 단일 코어 내에서 최대 1,800개의 병렬 입력 스트림을 수용할 수 있으며, 이는 코어 복제나 시간 순차적 스케줄링에 의존하는 기존 전자 프로세서가 달성하기 어려운 수준의 네이티브 병렬성입니다.
• 일반성: 본 연구는 현대 AI 워크로드의 데이터 이동 병목 현상을 극êm하기 위한 경로를 제공하며, 다른 아키텍처 및 재료 시스템으로 확장 가능한 일반적인 완전 다중화 광 컴퓨팅 프레임워크를 제공합니다.

저자들은 시스템이 모든 다중화된 채널에 대해 균일하게 프로그래밍된 계산을 보존한다는 점을 강조하며, 이는 이전의 다차원 시연들이 유지하기 어려워했던 고충실도 텐서 프로세싱의 핵심 요구사항입니다.