Expert Race: A Flexible Routing Strategy for Scaling Diffusion Transformer with Mixture of Experts
Created by
Haebom
Category
Empty
저자
Yike Yuan, Ziyu Wang, Zihao Huang, Defa Zhu, Xun Zhou, Jingyi Yu, Qiyang Min
개요
본 논문에서는 확산 트랜스포머를 위한 새로운 MoE(Mixture of Experts) 모델인 Race-DiT를 제안합니다. Race-DiT는 유연한 라우팅 전략인 Expert Race를 통해 토큰과 전문가 간의 경쟁을 허용하여 중요한 토큰에 전문가를 동적으로 할당합니다. 또한, 얕은 레이어 학습의 어려움을 해결하기 위한 레이어별 정규화와 모드 붕괴를 방지하기 위한 라우터 유사성 손실을 제안합니다. ImageNet에 대한 광범위한 실험을 통해 제안된 방법의 효과성을 검증하고, 성능 향상과 확장 가능성을 보여줍니다.
시사점, 한계점
•
시사점:
◦
확산 트랜스포머의 확장성 및 성능 향상에 기여하는 새로운 MoE 모델 Race-DiT 제시.
◦
Expert Race 라우팅 전략을 통해 중요 토큰에 대한 전문가 할당을 효율적으로 수행.