Paper ID | MLR-APPL-IVASR-5.5 | ||
Paper Title | DEEP REINFORCEMENT IMAGE MATCHING WITH SELF-TERMINATION | ||
Authors | Onkar Krishna, Go Irie, Xiaomeng Wu, Akisato Kimura, Kunio Kashino, NTT Corporation, Japan | ||
Session | MLR-APPL-IVASR-5: Machine learning for image and video analysis, synthesis, and retrieval 5 | ||
Location | Area C | ||
Session Time: | Tuesday, 21 September, 15:30 - 17:00 | ||
Presentation Time: | Tuesday, 21 September, 15:30 - 17:00 | ||
Presentation | Poster | ||
Topic | Applications of Machine Learning: Machine learning for image & video analysis, synthesis, and retrieval | ||
IEEE Xplore Open Preview | Click here to view in IEEE Xplore | ||
Abstract | Deep reinforcement learning-based image matching sequentially searches only the promising regions in the reference image that match the query, leading to a significantly small number of steps compared to traditional methods. Since existing methods do not have any function to judge whether the target region has been successfully identified or not, they continue to search until the preset maximum number of search steps is reached. In this paper, we propose a deep image matching network that can terminate the matching process by itself. Our network is designed to have a halting module that identifies whether the current reference region matches the query based on the image features and the search history. The entire network is effectively trained end-to-end in a framework of deep reinforcement learning that incorporates a new loss function to evaluate the accuracy of the termination decision. Experimental results demonstrate that our method can achieve highly competitive or better matching accuracy with fewer search steps than the existing methods. |