论文标题

蒸馏神经网络模型以检测和描述图像的关键点

Distillation of neural network models for detection and description of key points of images

论文作者

Yashchenko, A. V., Belikov, A. V., Peterson, M. V., Potapov, A. S.

论文摘要

图像匹配和分类方法以及同步位置和映射被广泛用于嵌入式和移动设备上。他们最大的资源密集型部分是对图像的要点的检测和描述。而且,如果可以在移动设备上实时执行经典检测和描述关键点的方法,那么对于具有最佳质量的现代神经网络方法,很难使用。因此,重要的是要提高神经网络模型的速度以检测和描述关键点。研究的主题是蒸馏作为减少神经网络模型的方法之一。术的目的是获得更紧凑的检测模型和关键点的描述,以及对获得此模型的过程的描述。测试了一种用于检测和描述关键点的神经网络的蒸馏方法。提出了在研究框架中提供最佳结果的目标函数和训练参数。已经引入了一个新的数据集,用于测试关键点检测方法,以及分配的关键点及其相应的本地功能的新质量指标。由于以描述方式训练,具有相同数量参数的新模型在比较关键点方面比原始模型更高。一个具有明显少数参数的新模型显示了匹配点的准确性,接近原始模型的准确性。

Image matching and classification methods, as well as synchronous location and mapping, are widely used on embedded and mobile devices. Their most resource-intensive part is the detection and description of the key points of the images. And if the classical methods of detecting and describing key points can be executed in real time on mobile devices, then for modern neural network methods with the best quality, such use is difficult. Thus, it is important to increase the speed of neural network models for the detection and description of key points. The subject of research is distillation as one of the methods for reducing neural network models. The aim of thestudy is to obtain a more compact model of detection and description of key points, as well as a description of the procedure for obtaining this model. A method for the distillation of neural networks for the task of detecting and describing key points was tested. The objective function and training parameters that provide the best results in the framework of the study are proposed. A new data set has been introduced for testing key point detection methods and a new quality indicator of the allocated key points and their corresponding local features. As a result of training in the described way, the new model, with the same number of parameters, showed greater accuracy in comparing key points than the original model. A new model with a significantly smaller number of parameters shows the accuracy of point matching close to the accuracy of the original model.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源