본문으로 건너뛰기

© 2026 Molayo

arXiv논문2026. 05. 20. 16:33

하드웨어 인지적 Transformer 적응을 위한 계층별 특화 스칼라 함수 진화

요약

Vision Transformers(ViT)를 엣지 디바이스에 배포할 때 발생하는 계층 정규화의 계산 복잡성과 병목 현상을 해결하기 위한 새로운 프레임워크를 제안합니다. 유전 프로그래밍(GP)을 통해 각 계층에 최적화된 이질적인 스칼라 함수를 진화시킴으로써, 모델의 전면적인 재학습 없이도 높은 정확도를 유지하며 하드웨어 효율성을 극대화합니다.

핵심 포인트

  • 계층 정규화(Layer Normalization)를 하드웨어 친화적인 스칼라 근사치로 교체하여 전역 리덕션 병목 현상 해결
  • 유전 프로그래밍(GP)을 활용하여 사전 학습된 가중치로부터 계층별로 특화된 스칼라 함수를 직접 진화
  • 사후 학습 재정렬(Post-training Re-alignment) 전략을 통해 모델 전체 재학습 없이 효율적인 최적화 가능
  • 기존 균질한 방식 대비 높은 분산 포착 능력($R^2$ 91.6%)과 엣지 가속기에서의 낮은 산술 복잡도 및 메모리 트래픽 달성

Vision Transformers (ViTs)는 까다로운 비전 작업에서 최첨단 성능을 달나하지만, 계층 정규화 (Layer Normalization)로 인해 발생하는 계산 복잡성과 전역 리덕션 병목 현상 (Global Reduction Bottleneck)으로 인해 엣지 디바이스 (Edge Devices)에 배포하는 데 심각한 제약을 받습니다. 최근의 방법들은 정규화 계층을 하드웨어 친화적인 스칼라 근사치 (Scalar Approximations)로 교체함으로써 이를 우회하려고 시도합니다. 그러나 이러한 균질한 (Homogeneous) 교체 방식은 모든 계층의 동작에 최적으로 부합하지 않으며, 비용이 많이 드는 모델 재학습 (Model Retraining)에 의존합니다. 본 연구에서는 유전 프로그래밍 (Genetic Programming, GP)을 활용하여 사전 학습된 가중치 (Pre-trained Weights)로부터 이질적이고 계층별로 특화된 스칼라 함수를 직접 진화시키는 매우 효율적인 하드웨어 인지적 프레임워크를 제안합니다. 새로운 사후 학습 재정렬 (Post-training Re-alignment) 전략과 결합된 우리의 접근 방식은 모델을 처음부터 완전히 재학습할 필요를 제거합니다. 우리가 진화시킨 수식들은 대상 정규화 동작을 정확하게 근사하며, 균질한 베이스라인 (Homogeneous Baselines)이 $70.2%$의 분산을 포착하는 것에 비해 $91.6%$의 분산 ($R^2$)을 포착합니다. 이를 통해 우리의 수정된 아키텍처는 단 20 에포크 (Epochs) 만에 ImageNet-1K의 Top-1 정확도의 $84.25%$를 회복할 수 있습니다. 전역 리덕션 병목 현상을 제거하면서도 이러한 성능을 유지함으로써, 우리의 접근 방식은 산술 복잡도 (Arithmetic Complexity)와 오프칩 메모리 트래픽 (Off-chip Memory Traffic) 사이의 매우 유리한 트레이드오프 (Trade-off)를 구축하며, 엣지 가속기 (Edge Accelerators)에서 ViT를 효율적으로 배포하는 데 있어 주요 장벽을 제거합니다.

AI 자동 생성 콘텐츠

본 콘텐츠는 arXiv cs.AR의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.

원문 바로가기
0

댓글

0