HyperAIHyperAI

Command Palette

Search for a command to run...

DISK: Learning local features with policy gradient

Tyszkiewicz Michał J. ; Fua Pascal ; Trulls Eduard

Abstract

Local feature frameworks are difficult to learn in an end-to-end fashion, dueto the discreteness inherent to the selection and matching of sparse keypoints.We introduce DISK (DIScrete Keypoints), a novel method that overcomes theseobstacles by leveraging principles from Reinforcement Learning (RL), optimizingend-to-end for a high number of correct feature matches. Our simple yetexpressive probabilistic model lets us keep the training and inference regimesclose, while maintaining good enough convergence properties to reliably trainfrom scratch. Our features can be extracted very densely while remainingdiscriminative, challenging commonly held assumptions about what constitutes agood keypoint, as showcased in Fig. 1, and deliver state-of-the-art results onthree public benchmarks.


Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing

HyperAI Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp