LOTUS: Improving Transformer Efficiency with Sparsity Pruning and Data Lottery Tickets

  • 2024-05-02 00:30:12
  • Ojasw Upadhyay
  • 0

Abstract

Vision transformers have revolutionized computer vision, but theircomputational demands present challenges for training and deployment. Thispaper introduces LOTUS (LOttery Transformers with Ultra Sparsity), a novelmethod that leverages data lottery ticket selection and sparsity pruning toaccelerate vision transformer training while maintaining accuracy. Our approachfocuses on identifying and utilizing the most informative data subsets andeliminating redundant model parameters to optimize the training process.Through extensive experiments, we demonstrate the effectiveness of LOTUS inachieving rapid convergence and high accuracy with significantly reducedcomputational requirements. This work highlights the potential of combiningdata selection and sparsity techniques for efficient vision transformertraining, opening doors for further research and development in this area.

 

Quick Read (beta)

loading the full paper ...