mlsanityData to 13 Oct 2026

Learning What to Predict: Downstream-Guided Task Design for Continued Pretraining

arXiv 2601.22108 ↗First posted 29 Jan 2026cs.LG cs.AIcited by 1

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

  1. 2020Language Models are Few-Shot Learnerscited by 32,320
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,656
  3. 2022Training language models to follow instructions with human feedbackcited by 15,856
  4. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,549
  5. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,009
  6. 2021Training Verifiers to Solve Math Word Problemscited by 9,916
  7. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,986
  8. 2020Measuring Massive Multitask Language Understandingcited by 7,876
  9. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,364
  10. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,915
  11. 2022Scalable Diffusion Models with Transformerscited by 5,973
  12. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,270
36 more
  1. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,870
  2. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,730
  3. 2017Deep reinforcement learning from human preferencescited by 3,871
  4. 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,765
  5. 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,850
  6. 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,082
  7. 2017Understanding Black-box Predictions via Influence Functionscited by 1,729
  8. 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,556
  9. 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,150
  10. 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
  11. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,018
  12. 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 954
  13. 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 730
  14. 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 572
  15. 2019Meta-Learning with Implicit Gradientscited by 523
  16. 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 359
  17. 2021Graph Contrastive Learning Automatedcited by 287
  18. 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
  19. 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
  20. 2018Revisiting Oxford and Paris: Large-Scale Image Retrieval Benchmarkingcited by 190
  21. 2025AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning datasetcited by 171
  22. 2024The pitfalls of next-token predictioncited by 160
  23. 2023Continual Pre-Training of Large Language Models: How to (re)warm your model?cited by 137
  24. 2019MLPerf Training Benchmarkcited by 114
  25. 2022Adversarial Masking for Self-Supervised Learningcited by 78
  26. 2017Learning the Structure of Generative Models without Labeled Datacited by 50
  27. 2022AdaMAE: Adaptive Masking for Efficient Spatiotemporal Learning with Masked Autoencoderscited by 41
  28. 2024MathCoder2: Better Math Reasoning from Continued Pretraining on Model-translated Mathematical Codecited by 39
  29. 2022Bringing Your Own View: Graph Contrastive Learning without Prefabricated Data Augmentationscited by 28
  30. 2024I Don't Know: Explicit Modeling of Uncertainty with an [IDK] Tokencited by 26
  31. 2025How Do LLMs Acquire New Knowledge? A Knowledge Circuits Perspective on Continual Pre-Trainingcited by 22
  32. 2019Meta-Surrogate Benchmarking for Hyperparameter Optimizationcited by 16
  33. 2022Stochastic smoothing of the top-K calibrated hinge loss for deep imbalanced classificationcited by 10
  34. 2020Exploring the limits of Concurrency in ML Training on Google TPUscited by 9
  35. 2021Language Modelling via Learning to Rankcited by 7
  36. 2024Pretraining with Random Noise for Fast and Robust Learning without Weight Transportcited by 5

Built on by

Cited by year (2026 to 13 Oct)

Citing papers per year: 2026: 1.

1 paper.