arXiv · 2411.01238
Efficient Sparse Training with Structured Dropout
Abstract
Dropout is a common regularisation technique in deep learning that improves generalisation. Even though it introduces sparsity and thus potential for higher throughput, it usually cannot bring speed-ups on GPUs due to its unstructured nature. In this project, I experiment with SparseDrop, a structured, hardware-friendly variant of dropout that can exploit such sparsity. I provide a CUDA implementation of SparseDrop, achieving speed-ups against its dense counterpart even at low sparsity levels. The empirical results demonstrate that SparseDrop provides similar, or sometimes even better, regularisation properties as standard dropout. This suggests its potential as a drop-in replacement to standard dropout with faster training speeds. The source code is available at https://github.com/andylolu2/sparse-dropout
Explore related subjects
Keep this discovery
Andy Lo. 2024-11-02. Efficient Sparse Training with Structured Dropout. https://arxiv.org/abs/2411.01238
Cite the original work for its findings. Save a collection to share your selection of sources.