atlas

Research map / Architectures and efficiency / RNNs and normalization

Adaptive Depth Networks with Skippable Sub-Paths

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 adaptive depth networks (ADN) that split each residual stage into a mandatory and a skippable sub-path, trained via a simple super-net-to-base-net self-distillation with skip-aware BNs/LNs, enabling combinatorial depth selection for CNNs and transformers.

Problem

Adaptive depth networks are hard and slow to train because prior methods self-distill exhaustively over many sub-networks, and there is no principled account of which layers can be skipped.

Approach

Divide each residual stage into a mandatory sub-path and a skippable sub-path; train only the super-net (teacher) and base-net (student) with KL self-distillation so the skippable blocks act as small feature refinements, then combine sub-paths at test time to get 2^N_r sub-networks.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add self-distillation and skip-aware BNs (ablation)ResNet50-ADN · ImageNet validationAcc@1: 75.2% to 77.6%; 72.2% to 76.1%Helped
Add self-distillation (ablation)ResNet50-ADN with skip-aware BNs · ImageNet validationAcc@1: 76.6% to 77.6%; 75.1% to 76.1%Helped
Add skip-aware BNs (ablation)ResNet50-ADN with self-distillation · ImageNet validationAcc@1: 76.1% to 77.6%; 74.9 % to 76.1%Helped
Add intermediate-feature distillation to output distillation (ablation)ResNet50-ADN · ImageNet validationAcc@1: 77.6% to 77.3%; 76.1% to 76.2%No clear effect
Use FFFF as teacher and TTTT as student instead of random teacher/student sampling (ablation)ResNet50-ADN · ImageNet validationAcc@1: 77.1 to 77.6; 76.7 to 77.1; 76.4 to 76.5; 75.5 to 76.0; 74.8 to 76.1Helped
Use FFFF as teacher and TTTT as student instead of random teacher/student sampling (ablation)ViT-b/32-ADN · ImageNet validationAcc@1: 75.2 to 76.6; 74.7 to 76.0; 73.9 to 75.5; 72.9 to 74.5; 71.1 to 74.3Helped
Make mandatory sub-paths shorter than skippable sub-paths (ablation)ResNet50-ADN · ImageNet validationAcc@1Hurt
Increase mandatory sub-path lengths beyond the default ratio (ablation)ResNet50-ADN · ImageNet validationAcc@1No clear effect
Make mandatory sub-paths shorter by selecting the last two encoder blocks as skippable (ablation)ViT-b/16-ADN · ImageNet validationAcc@1Hurt
Use adaptive depth sub-networks instead of individually trained networksResNet50 · ImageNet validationAcc@1Helped
Reduce FLOPs through depth adaptationResNet50-ADN on Jetson Orin Nano · On-device inferenceinference latency: reducing FLOPs by 38% ... reduces ... inference latency ... by 35%Helped
Use ResNet50-ADN as the detector backbone rather than individual ResNet50Faster R-CNN · MS COCO val2017Box AP: 36.4 to 37.8Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet ILSVRC2012 MS COCO 2017

Benchmarks: ImageNet ILSVRC2012 validation MS COCO val2017

Metrics: Top-1 accuracy (Acc@1) FLOPs Box AP Mask AP inference latency energy consumption

Baselines: ResNet50 (individual) ResNet50 (KD individual) AlphaNet S-ResNet50 GReg-1 DepGraph DR-ResNet50 AlphaNet-0.75x X-Pruner SPViT-Swin-Ti Ada-DeiT-S WDPruning

Builds on: ResNet Swin Transformer AlphaNet DeiT training recipe

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