Efficient Compilation Method for Remote Sensing Deep Learning Models Based on Search Optimization and Adaptive Clustering

Deploying deep learning-based remote sensing image interpretation models in orbit can alleviate data downlink pressure and enhance processing efficiency. To adapt to the arithmetic conditions of on-orbit platforms, deep learning models must undergo compilation before deployment, where the search and measurement of operator scheduling are critical aspects. The existing compilation methods related to the Tensor Virtual Machine (TVM) compiler suffer from inefficiencies caused by suboptimal starting point selection and extensive search spaces during the search process. Additionally, the clustering algorithm employed during the measurement of candidate scheduling relies on fixed dimensions and lacks flexible application management, resulting in slow convergence. This paper proposes corresponding optimization methods built upon the TVM framework to address these two challenges. In the search aspect, techniques such as starting point planning, salient reduction, and model sharing considerably reduce search time. In the measurement aspect, the clustering process is optimized through adaptive scheduling management and dimension adaptation, enhancing measurement efficiency. Experimental results indicate that, compared to current state-of-the-art deep learning compilers, the proposed method increases compilation speed by an average of 2.6 times while maintaining consistent inference speed.