Lightweight tensorial convolutional neural network for lunar impact crater detection

Abstract

Impact crater detection is one of the most important and critical studies of the geological history of planetary surfaces. Recent literature on crater identification demonstrates the powerful potential of deep-learning algorithms. However, these methods potentially assumed that the detection is conducted in devices with sufficient computing resources, and used networks with massive learnable parameters to obtain superb performance. In this paper, we proposed a lightweight tensorial convolutional neural network (CNN) for lunar impact crater detection. Specifically, we develop a tensorial you only look once (TYOLO) network by representing the convolutional kernels in tensor train (TT) decomposition. Thus, the parameter redundancy of TYOLO can be greatly reduced. To the best of our knowledge, this is the first work that considers the lightweight CNN model for impact crater detection. We demonstrate the lunar impact crater detection results over the ChangE-2 lunar digital orthophotos model (DOM) and digital evaluated model (DEM). Experimental results show that the proposed TYOLO network can obtain 1.44×speed up and 39× compression with 2.4% improvement in crater detection. Overall, the proposed network demonstrates great potential in spacecraft with emerging computing resource constraints.