Semantic-DARTS: Elevating Semantic Learning for Mobile Differentiable Architecture Search

Bicheng Guo,Shibo He,Miaojing Shi,Kaicheng Yu,Jiming Chen,Xuemin Shen
DOI: https://doi.org/10.1109/jiot.2024.3462954
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:Differentiable ARchitecture Search (DARTS) is a prevailing direction in automatic machine learning, but it may suffer from performance collapse and generalization issues. Recent efforts mitigate them by integrating regularization into architectural parameters or rule-based operations selection. These efforts primarily emphasize learning the global class-specific features through the image classification task, while overlooking the fine-grained local information during the search process. In this paper, we take the first trial to observe that three semantic challenges arise from the classification-based DARTS: i) inaccurate class-specific features; ii) partial target attention; and iii) blurred semantic regions. To tackle them in one shot, we propose Semantic-DARTS, combining the Masked Image Modeling (MIM) paradigm with the classification task to incorporate local semantic information into the architecture search. Specifically, we design a lightweight reconstruction head that recovers the corrupted image based on the condensed latent feature, which learns both the local semantics and their relationship patch-wisely. Simultaneously, the concurrent classification head strengthens the connection between the global category of the target and the local semantics of their parts. As evidenced by our experiments, the proposed approach achieves state-of-the-art results on CIFAR-10, CIFAR-100, and ImageNet. Furthermore, the searched model is not only able to improve global class-specific features, but also to capture fine-grained local representations, improving both the classification performance and the generalization ability.
What problem does this paper attempt to address?