Header menu link for other important links
X
Learning Modular Structures That Generalize Out-of-Distribution (Student Abstract)
A. Ashok, C. Devaguptapu,
Published in Association for the Advancement of Artificial Intelligence
2022
Volume: 36
   
Pages: 12905 - 12906
Abstract
Out-of-distribution (O.O.D.) generalization remains to be a key challenge for real-world machine learning systems. We describe a method for O.O.D. generalization that, through training, encourages models to only preserve features in the network that are well reused across multiple training domains. Our method combines two complementary neuron-level regularizers with a probabilistic differentiable binary mask over the network, to extract a modular sub-network that achieves better O.O.D. performance than the original network. Preliminary evaluation on two benchmark datasets corroborates the promise of our method. Copyright © 2022, Association for the Advancement of Artificial Intelligence (www.aaai.org). All rights reserved.
About the journal
JournalProceedings of the 36th AAAI Conference on Artificial Intelligence, AAAI 2022
PublisherAssociation for the Advancement of Artificial Intelligence