AutoTailor: Automatic and Efficient Adaptive Model Deployment for Diverse Edge Devices
Organizations: Southeast University · Hong Kong University of Science and Technology · Tsinghua University · University of Macau
Abstract
On-device machine learning (ML) has become a fundamental component of emerging mobile applications. Adaptive model deployment delivers efficient inference for heterogeneous device capabilities and performance requirements through customizing neural architectures. SuperNet-based approaches offer a promising solution by generating numerous model variants from a pre-trained ML model. However, applying SuperNet in existing frameworks requires tedious model-aware development and time-consuming hardware-aware profiling, which limits their practical adoption. We present AutoTailor, the first framework to enable automated, end-to-end SuperNet-based adaptive model deployment for edge devices. Unlike manual SuperNet construction, AutoTailor employs a computation graph-guided compilation approach to automatically transform user-provided ML models into SuperNets. To support efficient specialization, AutoTailor incorporates learning-free latency and accuracy predictors, enabling low-cost yet accurate performance prediction. Our extensive evaluations demonstrate that AutoTailor reduces the lines of code for SuperNet construction by 10.8--34.5, reduces predictor profiling time by 2.2--11.5, and achieves up to 15.60% absolute accuracy improvement and 60.03% latency reduction compared to state-of-the-art approaches across diverse models and devices. AutoTailor's source code is publicly available at https://github.com/liu-mengyang/AutoTailor.
Figures & tables
| Model | ResNet50 | MobileNetv3 |
|---|---|---|
| Number of variants | ||
| Latency profiling (s/variant) | ||
| Accuracy profiling (s/variant) |
| Models | normal | + space pruning | + TailorIR |
| (GPU hours) | (GPU hours) | (CPU minutes) | |
| ResNet50 | 1.25E+09 | 10.8 | 11.8 |
| MobileNetv2 | 872000 | 0.0382 | 2.2 |
| ConvNext | 8.08978E+15 | 0.0259 | 4.8 |
| LeViT | 1.02E+41 | 57.44 | 46.5 |
| Original DNN | # of Modifications | # of Variants |
|---|---|---|
| ResNet50 He et al. (2016) | 100 | |
| MobileNetv2 Sandler et al. (2018) | 85 | |
| MobileNetv3 Howard et al. (2019) | 160 | |
| ProxylessNet Cai et al. (2019) | 166 | |
| ConvNext-Tiny Liu et al. (2022) | 163 | |
| ViT-Base Dosovitskiy et al. (2021) | 103 |
Edge ML
Edge machine learning runs and adapts models on nearby devices to reduce latency, bandwidth use, and exposure of sensitive data.
See all 129 papers