mlsanityData to 14 Oct 2026

Learning from Oblivion: Predicting Knowledge Overflowed Weights via Retrodiction of Forgetting

arXiv 2508.05059 ↗First posted 7 Aug 2025cs.LG cs.AI cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Pre-trained weights have become a cornerstone of modern deep learning, enabling efficient knowledge transfer and improving downstream task performance, especially in data-scarce scenarios. However, a fundamental question remains: how can we obtain better pre-trained weights that encapsulate more knowledge beyond the given dataset? In this work, we introduce KNowledge-Overflowed Weights (KNOW) prediction, a novel strategy that leverages structured forgetting and its inversion to synthesize knowledge-enriched weights. Our key insight is that sequential fine-tuning on progressively downsized datasets induces a structured forgetting process, which can be modeled and reversed to recover knowledge as if trained on a larger dataset. We construct a dataset of weight transitions governed by this controlled forgetting and employ meta-learning to model weight prediction effectively. Specifically, our KNowledge-Overflowed Weights Nowcaster (KNOWN) acts as a hyper-model that learns the general evolution of weights and predicts enhanced weights with improved generalization. Extensive experiments across diverse datasets and architectures demonstrate that KNOW prediction consistently outperforms Naive fine-tuning and simple weight prediction, leading to superior downstream performance. Our work provides a new perspective on reinterpreting forgetting dynamics to push the limits of knowledge transfer. The code and pre-trained model are available at https://github.com/jjh6297/KNOW

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 47 of its 64 references that are in the map. First posted on arXiv 7 Aug 2025.

Builds on

47 of 64 references.

References · 47

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  3. 2016Densely Connected Convolutional Networkscited by 6,720
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
  6. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  7. 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
  8. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  9. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
  10. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  11. 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,223
  12. 2017Gradient Episodic Memory for Continual Learningcited by 1,898
35 more
  1. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  2. 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
  3. 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
  4. 2022Editing Models with Task Arithmeticcited by 1,120
  5. 2017Deeper, Broader and Artier Domain Generalizationcited by 986
  6. 2017Revisiting Unreasonable Effectiveness of Data in Deep Learning Eracited by 919
  7. 2021PVT v2: Improved Baselines with Pyramid Vision Transformercited by 751
  8. 2013An Empirical Investigation of Catastrophic Forgetting in Gradient-Based Neural Networkscited by 722
  9. 2023An Empirical Study of Catastrophic Forgetting in Large Language Models During Continual Fine-tuningcited by 639
  10. 2021Merging Models with Fisher-Weighted Averagingcited by 621
  11. 2016Learning to learn by gradient descent by gradient descentcited by 620
  12. 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 618
  13. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
  14. 2018Do Better ImageNet Models Transfer Better?cited by 565
  15. 2018Rethinking ImageNet Pre-trainingcited by 377
  16. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
  17. 2016What makes ImageNet good for transfer learning?cited by 192
  18. 2022Patching open-vocabulary models by interpolating weightscited by 190
  19. 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
  20. 2024Localizing Task Information for Improved Model Merging and Compressioncited by 131
  21. 2021Overcoming Catastrophic Forgetting in Incremental Few-Shot Learning by Finding Flat Minimacited by 118
  22. 2017Learned Optimizers that Scale and Generalizecited by 110
  23. 2021Exploring the Limits of Large Scale Pre-trainingcited by 104
  24. 2024Model Stock: All we need is just a few fine-tuned modelscited by 92
  25. 2021Loss Surface Simplexes for Mode Connecting Volumes and Fast Ensemblingcited by 75
  26. 2024MagMax: Leveraging Model Merging for Seamless Continual Learningcited by 67
  27. 2025Overcoming catastrophic forgetting in neural networkscited by 55
  28. 2017Learning Gradient Descent: Better Generalization and Longer Horizonscited by 50
  29. 2021GradInit: Learning to Initialize Neural Networks for Stable and Efficient Trainingcited by 44
  30. 2023Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?cited by 29
  31. 2020Embedded Encoder-Decoder in Convolutional Networks Towards Explainable AIcited by 24
  32. 2024Accelerating Training with Neuron Interaction and Nowcasting Networkscited by 12
  33. 2022Towards Theoretically Inspired Neural Initialization Optimizationcited by 11
  34. 2017Introspection: Accelerating Neural Network Training By Learning Weight Evolutioncited by 4
  35. 2022Optimizer Amalgamationcited by 1

Built on by

Not cited yet.