Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Composed Fine-Tuning: Freezing Pre-Trained Denoising Autoencoders for Improved Generalization
(Submitted on 29 Jun 2020 (v1), last revised 24 Oct 2023 (this version, v4))
Abstract: We focus on prediction problems with structured outputs that are subject to output validity constraints, e.g. pseudocode-to-code translation where the code must compile. While labeled input-output pairs are expensive to obtain, "unlabeled" outputs, i.e. outputs without corresponding inputs, are freely available (e.g. code on GitHub) and provide information about output validity. We can capture the output structure by pre-training a denoiser to denoise corrupted versions of unlabeled outputs. We first show that standard fine-tuning after pre-training destroys some of this structure. We then propose composed fine-tuning, which fine-tunes a predictor composed with the pre-trained denoiser, which is frozen to preserve output structure. For two-layer ReLU networks, we prove that composed fine-tuning significantly reduces the complexity of the predictor, thus improving generalization. Empirically, we show that composed fine-tuning improves over standard fine-tuning on two pseudocode-to-code translation datasets (3% and 6% relative). The improvement from composed fine-tuning is magnified on out-of-distribution (OOD) examples (4% and 25% relative).
Submission history
From: Sang Michael Xie [view email][v1] Mon, 29 Jun 2020 17:14:35 GMT (739kb,D)
[v2] Tue, 10 Nov 2020 06:36:18 GMT (875kb,D)
[v3] Fri, 11 Jun 2021 23:45:47 GMT (3005kb,D)
[v4] Tue, 24 Oct 2023 23:44:37 GMT (6005kb,D)
Link back to: arXiv, form interface, contact.