Learning What to Predict: Downstream-Guided Task Design for Continued Pretraining
Not on the map. Why?
Abstract (arXiv)
Continued pretraining is optimized with fixed self-supervised tasks but selected by downstream performance, creating a coarse feedback loop in which practitioners evaluate checkpoints, change data mixtures or objectives, and restart runs, while individual updates remain blind to target capabilities. We ask whether a small set of verifiable downstream examples can provide step-level feedback without directly supervising the learner. We introduce V-pretraining, which decouples a learner trained only with a self-supervised loss from a lightweight task designer that constructs targets or views for unlabeled batches. Given the current learner and batch, V-pretraining scores a candidate construction by predicting the first-order reduction in downstream loss after the induced self-supervised update. The designer maximizes this value; the learner then applies the update with targets or views detached, so downstream labels never update learner parameters. We instantiate V-pretraining as adaptive top-K soft targets for language modeling and learned views or masks for self-supervised vision. Across both modalities, V-pretraining improves target capabilities without degrading generalization. Under wall-clock-matched continued pretraining, it improves GSM8K Pass@1 for Qwen models using 1,024 GSM8K examples only as feedback, including a +7.4 point single-run gain for Qwen2.5-0.5B. In vision, it improves DINOv3 transfer to ADE20K semantic segmentation and NYUv2 depth estimation while preserving ImageNet linear accuracy, suggesting that feedback-guided task construction can improve target capabilities without collapsing general-purpose representations.
Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Builds on 48 of its 63 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
48 of 63 references.
References · 48
- 2020Language Models are Few-Shot Learnerscited by 32,320
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,656
- 2022Training language models to follow instructions with human feedbackcited by 15,856
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,549
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,009
- 2021Training Verifiers to Solve Math Word Problemscited by 9,916
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,986
- 2020Measuring Massive Multitask Language Understandingcited by 7,876
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,364
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,915
- 2022Scalable Diffusion Models with Transformerscited by 5,973
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,270
36 more
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,870
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,730
- 2017Deep reinforcement learning from human preferencescited by 3,871
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,765
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,850
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,082
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,729
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,556
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,150
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,018
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 954
- 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 730
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 572
- 2019Meta-Learning with Implicit Gradientscited by 523
- 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 359
- 2021Graph Contrastive Learning Automatedcited by 287
- 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
- 2018Revisiting Oxford and Paris: Large-Scale Image Retrieval Benchmarkingcited by 190
- 2025AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning datasetcited by 171
- 2024The pitfalls of next-token predictioncited by 160
- 2023Continual Pre-Training of Large Language Models: How to (re)warm your model?cited by 137
- 2019MLPerf Training Benchmarkcited by 114
- 2022Adversarial Masking for Self-Supervised Learningcited by 78
- 2017Learning the Structure of Generative Models without Labeled Datacited by 50
- 2022AdaMAE: Adaptive Masking for Efficient Spatiotemporal Learning with Masked Autoencoderscited by 41
- 2024MathCoder2: Better Math Reasoning from Continued Pretraining on Model-translated Mathematical Codecited by 39
- 2022Bringing Your Own View: Graph Contrastive Learning without Prefabricated Data Augmentationscited by 28
- 2024I Don't Know: Explicit Modeling of Uncertainty with an [IDK] Tokencited by 26
- 2025How Do LLMs Acquire New Knowledge? A Knowledge Circuits Perspective on Continual Pre-Trainingcited by 22
- 2019Meta-Surrogate Benchmarking for Hyperparameter Optimizationcited by 16
- 2022Stochastic smoothing of the top-K calibrated hinge loss for deep imbalanced classificationcited by 10
- 2020Exploring the limits of Concurrency in ML Training on Google TPUscited by 9
- 2021Language Modelling via Learning to Rankcited by 7
- 2024Pretraining with Random Noise for Fast and Robust Learning without Weight Transportcited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Crypto x AI, AI x Crypto: A Surveycited by 1
- 2026Crypto x AI, AI x Crypto: A Survey11 Jun 2026