We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Towards Optimization and Model Selection for Domain Generalization: A Mixup-guided Solution

Abstract: The distribution shifts between training and test data typically undermine the performance of models. In recent years, lots of work pays attention to domain generalization (DG) where distribution shifts exist, and target data are unseen. Despite the progress in algorithm design, two foundational factors have long been ignored: 1) the optimization for regularization-based objectives, and 2) the model selection for DG since no knowledge about the target domain can be utilized. In this paper, we propose Mixup guided optimization and selection techniques for DG. For optimization, we utilize an adapted Mixup to generate an out-of-distribution dataset that can guide the preference direction and optimize with Pareto optimization. For model selection, we generate a validation dataset with a closer distance to the target distribution, and thereby it can better represent the target data. We also present some theoretical insights behind our proposals. Comprehensive experiments demonstrate that our model optimization and selection techniques can largely improve the performance of existing domain generalization algorithms and even achieve new state-of-the-art results.
Comments: Accepted by SIAM International Conference on Data Mining (SDM) 2024
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2209.00652 [cs.LG]
  (or arXiv:2209.00652v2 [cs.LG] for this version)

Submission history

From: Jindong Wang [view email]
[v1] Thu, 1 Sep 2022 02:18:00 GMT (445kb,D)
[v2] Thu, 4 Jan 2024 01:41:08 GMT (207kb,D)

Link back to: arXiv, form interface, contact.