共 50 条
Fine-Tuning DARTS for Image Classification
被引:31
|作者:
Tanveer, Muhammad Suhaib
[1
]
Khan, Muhammad Umar Karim
[2
]
Kyung, Chong-Min
[1
]
机构:
[1] Korea Adv Inst Sci & Technol KAIST, Sch Elect Engn, Daejeon 34141, South Korea
[2] Korea Adv Inst Sci & Technol, Ctr Integrated Smart Sensors CISS, Daejeon 34141, South Korea
关键词:
D O I:
10.1109/ICPR48806.2021.9412221
中图分类号:
TP18 [人工智能理论];
学科分类号:
081104 ;
0812 ;
0835 ;
1405 ;
摘要:
Neural Architecture Search (NAS) has gained attraction due to superior classification performance. Differential Architecture Search (DARTS) is a computationally light method. To limit computational resources DARTS makes numerous approximations. These approximations result in inferior performance. We propose to fine-tune DARTS using fixed operations as they are independent of these approximations. Our method offers a good trade-off between the number of parameters and classification accuracy. Our approach improves the top-1 accuracy on Fashion-MNIST, Comp Cars, and MIO-TCD datasets by 0.56%, 0.50%, and 0.39%, respectively compared to the state-of-the-art approaches. Our approach performs better than DARTS, improving the accuracy by 0.28%, 1.64%, 034%, 4.5%, and 3.27% compared to DARTS, on CIFAR-10, CIFAR-100, Fashion-MNIST, CompCars, and MIO-TCD datasets, respectively.
引用
收藏
页码:4789 / 4796
页数:8
相关论文