DPP-Net: Device-aware Progressive Search for Pareto-optimal Neural Architectures

By Jin-Dong Dong, An-Chieh Cheng, Da-Cheng Juan, Wei Wei, Min Sun. 2018

197 citations recorded by the source corpus.

Recent breakthroughs in Neural Architectural Search (NAS) have achieved state-of-the-art performances in applications such as image classification and language modeling. However, these techniques typically ignore device-related objectives such as inference time, memory usage, and power consumption. Optimizing neural architecture for device-related objectives is immensely crucial for deploying deep networks on portable devices with limited computing resources. We propose DPP-Net: Device-aware Progressive Search for Pareto-optimal Neural Architectures, optimizing for both device-related (e.g., inference time and memory usage) and device-agnostic (e.g., accuracy and model size) objectives. DPP-Net employs a compact search space inspired by current state-of-the-art mobile CNNs, and further improves search efficiency by adopting progressive search (Liu et al. 2017). Experimental results on CIFAR-10 are poised to demonstrate the effectiveness of Pareto-optimal networks found by DPP-Net, for three different devices: (1) a workstation with Titan X GPU, (2) NVIDIA Jetson TX1 embedded system, and (3) mobile phone with ARM Cortex-A53. Compared to CondenseNet and NASNet (Mobile), DPP-Net achieves better performances: higher accuracy & shorter inference time on various devices. Additional experimental results show that models found by DPP-Net also achieve considerably-good performance on ImageNet as well.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Efficiency & Compression, Neural Architecture Search & AutoML

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Alan L. Yuille
  2. Jiashi Feng
  3. Quoc V. Le
  4. Shuicheng Yan
  5. Yi Yang
  6. Liang-Chieh Chen
  7. Dacheng Tao
  8. Ming-Hsuan Yang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Jiashi Feng
  3. Alan L. Yuille
  4. Ming-Hsuan Yang
  5. Dacheng Tao
  6. Trevor Darrell
  7. Xiaogang Wang
  8. Yi Yang

In Efficiency & Compression

  1. Jiashi Feng
  2. Diana Marculescu
  3. Shuicheng Yan
  4. Zechun Liu
  5. Zichao Guo
  6. Xiangxiang Chu
  7. Ruijun Xu
  8. Haoyuan Mu

In Neural Architecture Search & AutoML

  1. Martin Wistuba
  2. Xiangxiang Chu
  3. Frank Hutter
  4. Xiaojun Chang
  5. Tejaswini Pedapati
  6. Ambrish Rawat
  7. Thomas Elsken
  8. Ruijun Xu

Citations in the corpus