Timezone: »

 
Submodular Batch Selection for Training Deep Neural Networks
Vineeth N Balasubramanian

Fri Jun 14 11:40 AM -- 12:00 PM (PDT) @ None

Mini-batch gradient descent based methods are the de facto algorithms for training neural network architectures today. We introduce a mini-batch selection strategy based on submodular function maximization. Our novel submodular formulation captures the informativeness of each sample and diversity of the whole subset. We design an efficient, greedy algorithm which can give high-quality solutions to this NP-hard combinatorial optimization problem. Our extensive experiments on standard datasets show that the deep models trained using the proposed batch selection strategy provide better generalization than Stochastic Gradient Descent as well as a popular baseline sampling strategy across different learning rates, batch sizes, and distance metrics.

Author Information

Vineeth N Balasubramanian (Indian Institute of Technology, Hyderabad)

More from the Same Authors

  • 2022 Poster: Matching Learned Causal Effects of Neural Networks with Domain Priors »
    Gowtham Reddy Abbavaram · Sai Srinivas Kancheti · Vineeth N Balasubramanian · Amit Sharma
  • 2022 Spotlight: Matching Learned Causal Effects of Neural Networks with Domain Priors »
    Gowtham Reddy Abbavaram · Sai Srinivas Kancheti · Vineeth N Balasubramanian · Amit Sharma
  • 2019 : Spotlight »
    Tyler Scott · Kiran Koshy · Jonathan Aigrain · Rene Bidart · Priyadarshini Panda · Dian Ang Yap · Yaniv Yacoby · Raphael Gontijo Lopes · Alberto Marchisio · Erik Englesson · Wanqian Yang · Moritz Graule · Yi Sun · Daniel Kang · Mike Dusenberry · Min Du · Hartmut Maennel · Kunal Menda · Vineet Edupuganti · Luke Metz · David Stutz · Vignesh Srinivasan · Timo Sämann · Vineeth N Balasubramanian · Sina Mohseni · Rob Cornish · Judith Butepage · Zhangyang Wang · Bai Li · Bo Han · Honglin Li · Maksym Andriushchenko · Lukas Ruff · Meet P. Vadera · Yaniv Ovadia · Sunil Thulasidasan · Disi Ji · Gang Niu · Saeed Mahloujifar · Aviral Kumar · SANGHYUK CHUN · Dong Yin · Joyce Xu Xu · Hugo Gomes · Raanan Rohekar
  • 2019 Poster: Neural Network Attributions: A Causal Perspective »
    Aditya Chattopadhyay · Piyushi Manupriya · Anirban Sarkar · Vineeth N Balasubramanian
  • 2019 Oral: Neural Network Attributions: A Causal Perspective »
    Aditya Chattopadhyay · Piyushi Manupriya · Anirban Sarkar · Vineeth N Balasubramanian