From Uniform to Adaptive: General Skip-Block Mechanisms for Efficient PDE Neural Operators
Created by
Haebom
Category
Empty
저자
Lei Liu, Zhongyi Yu, Hong Wang, Huanshuo Dong, Haiyang Xin, Hongwei Zhao, Bin Li
개요
Transformer 기반 신경 연산자를 위한 Skip-Block Routing (SBR) 프레임워크를 제안합니다. SBR은 복잡성 순위를 학습하여 추론 시 복잡한 토큰에 더 많은 처리 능력을 집중함으로써 계산 비용을 줄입니다. 특히, 물리적 필드의 복잡성이 다른 대규모 엔지니어링 작업에서 계산 효율성을 개선하는 것을 목표로 합니다. 실험 결과, SBR은 다양한 신경 연산자에 통합될 수 있으며, 정확도를 유지하면서 FLOPS를 약 50% 절감하고 최대 2배 빠른 추론 속도를 달성했습니다.