Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Probing transfer learning with a model of synthetic correlated datasets
 
research article

Probing transfer learning with a model of synthetic correlated datasets

Gerace, Federica  
•
Saglietti, Luca  
•
Sarao Mannelli, Stefano
Show more
January 26, 2022
Machine Learning: Science and Technology

Transfer learning can significantly improve the sample efficiency of neural networks, by exploiting the relatedness between a data-scarce target task and a data-abundant source task. Despite years of successful applications, transfer learning practice often relies on ad-hoc solutions, while theoretical understanding of these procedures is still limited. In the present work, we re-think a solvable model of synthetic data as a framework for modeling correlation between data-sets. This setup allows for an analytic characterization of the generalization performance obtained when transferring the learned feature map from the source to the target task. Focusing on the problem of training two-layer networks in a binary classification setting, we show that our model can capture a range of salient features of transfer learning with real data. Moreover, by exploiting parametric control over the correlation between the two data-sets, we systematically investigate under which conditions the transfer of features is beneficial for generalization.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

Gerace+et+al_2022_Mach._Learn. _Sci._Technol._10.1088_2632-2153_ac4f3f.pdf

Type

Postprint

Version

Accepted version

Access type

openaccess

License Condition

CC BY

Size

1.12 MB

Format

Adobe PDF

Checksum (MD5)

1c37fd9eaa27dfcb3f1b58944240b435

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés