atlas

Research map / Architectures and efficiency / RNNs and normalization

Dynamic Neural Regeneration: Enhancing Deep Learning Generalization on Small Datasets

NeurIPS 2024 · Architectures and efficiency › RNNs and normalization · approach: Efficient network compression and training · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Dynamic Neural Regeneration (DNR), an iterative training method that uses SNIP-based data-aware dynamic masking each generation to choose which parameters to reinitialize, improving generalization on small datasets.

Problem

Training deep networks on small datasets leads to overfitting; existing iterative weight-reinitialization methods (e.g., Knowledge Evolution) use random, fixed masks that may be suboptimal.

Approach

Replace KE's fixed random fit/reset mask with a mask recomputed at the start of each generation using SNIP connection-sensitivity on a data subset, reinitializing a fixed 20% of parameters with the lowest saliency.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace a fixed random mask with data-aware dynamic SNIP maskingDNR versus KE · small datasets (CUB-200-2011, FGVC-Aircraft, Stanford Dogs, Flower102, MIT67)accuracy: DNR (f10): 59.72 ±0.21 vs KE (f10): 58.11 ±0.25 (CUB); 55.87 ±0.47 vs 53.21 ±0.43 (Aircraft); 65.76±0.13 vs 64.56±0.31 (Helped
use DNR instead of the equal-computation long baselineResNet50 · CIFAR10accuracy: 93.60±0.29 -> 94.61±0.30Helped
use DNR instead of the equal-computation long baselineResNet50 · CIFAR100accuracy: 74.21±0.28 -> 75.05±0.23Helped
use DNR instead of the equal-computation long baselineResNet50 · Tiny-ImageNetaccuracy: 51.16 ±0.21 -> 54.50 ±0.26Helped
use DNR instead of vanilla transfer learningResNet18 · CUB-200-2011accuracy: 65.63 ±0.21 -> 68.56 ±0.24Helped
use SNIP instead of random selection (ablation)DNR · CUB-200-2011accuracy: 66.51±0.07 -> 71.87 ±0.22Helped
use SNIP instead of Fisher Importance (ablation)DNR · CUB-200-2011accuracy: 67.73 ±0.28 -> 71.87 ±0.22Helped
use SNIP instead of weight magnitude (ablation)DNR · CUB-200-2011accuracy: 64.18 ±0.19 -> 71.87 ±0.22Helped
vary the reinitialized parameter fraction from 5% to 40% (ablation)DNR · FGVC-Aircraft and CUB-200-2011accuracy: Aircraft: 65.34, 66.10, 66.63, 64.13, 62.79; CUB: 69.95, 70.15, 71.37, 68.42, 66.87Mixed
reduce importance-estimation data from 0.2 |D| to 128 samples (ablation)DNR · FGVC-Aircraft and CUB-200-2011accuracy: Aircraft: 66.63 -> 66.45; CUB: 71.37 -> 71.26No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: CUB-200-2011 Flower-102 MIT67 Stanford Dogs FGVC-Aircraft CIFAR-10 CIFAR-100 Tiny-ImageNet CIFAR-10-C PAPILA

Benchmarks: CUB-200-2011 Flower-102 MIT67 Stanford Dogs FGVC-Aircraft CIFAR-10 CIFAR-100 Tiny-ImageNet CIFAR-10-C PGD-10 adversarial attack

Metrics: accuracy mean corruption accuracy adversarial accuracy balanced accuracy sensitivity specificity

Baselines: Cross-entropy (long baseline) DSD BAN Knowledge Evolution (KE) LW LLF Label smoothing Transfer learning (vanilla fine-tuning)

Builds on: Knowledge Evolution SNIP Born Again Networks Dense-Sparse-Dense Forget and relearn (LLF)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy