Enhancing Transformer Efficiency Through Active Learning and Knowledge Distillation
摘要
In the recent years, Transformers have emerged as some of the most powerful models for sequential data processing, demonstrating exceptional performance across various domains. Their ability to capture long-term dependencies is remarkable and has given profound inference and detection capabilities. However, their computational demands for training and inference pose significant challenges. This study introduces two optimization techniques for transformers: uncertainty sampling-based active learning to enhance robustness and accuracy, and student-teacher knowledge distillation to improve efficiency. Experimental results on the Internet Movie databases dataset indicate a 7 \(\times \) speedup in inference time with an acceptable trade-off in accuracy, showcasing the effectiveness of our approach.