Presentation information

Organized Session

Organized Session » OS-3

[3J3-OS-3a] AutoML(自動機械学習)(1/2)

Thu. Jun 16, 2022 1:30 PM - 3:10 PM Room J (Room J)

オーガナイザ:大西 正輝(産業技術総合研究所)[現地]、日野 英逸(統計数理研究所/理化学研究所)

2:50 PM - 3:10 PM

[3J3-OS-3a-05] Model Size Constrained Optimization of DARTS in Neural Architecture Search

〇Kazuki Henmi1, Masaki Onishi2 (1. University of Tsukuba, 2. National Institute of Advanced Industrial Science and Technology)

Keywords:AutoML, Neural Architecture Search, Constrained Optimization

Deep learning, a machine learning method, has been applied in a variety of fields such as natural language processing and image recognition due to its high performance.
AutoML, a method to automate machine learning, has been widely studied, and Neural Architecture Search (NAS), a method to automatically optimize neural network models according to data and objectives, plays a very important role. NAS can find a very accurate model using DARTS, which uses the gradient method to search.
Generally, DARTS only optimizes accuracy, which improves recognition accuracy but also increases the amount of memory needed for the model. However, there is a limit to the quantity of memory that can be loaded when using deep learning on mobile devices and embedded systems. In this paper, we propose a method to search for a network model that considers accuracy and model size by adding constraints to DARTS. As a result, the proposed method enables us to search for network models with high accuracy in constraint conditions.

Authentication for paper PDF access

A password is required to view paper PDFs. If you are a registered participant, please log on the site from Participant Log In.
You could view the PDF with entering the PDF viewing password bellow.