Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. DISK: learning local features with policy gradient
 
conference paper

DISK: learning local features with policy gradient

Tyszkiewicz, Michal Jan  
•
Fua, Pascal  
•
Trulls, Eduard
June 1, 2020
Advances in Neural Information Processing Systems
Conference on Neural Information Processing Systems (NeurIPS)

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

  • Files
  • Details
  • Metrics
Type
conference paper
Author(s)
Tyszkiewicz, Michal Jan  
Fua, Pascal  
Trulls, Eduard
Date Issued

2020-06-01

Published in
Advances in Neural Information Processing Systems
Total of pages

9

Volume

33

Subjects

Computer vision

•

Local features

•

Image localization

•

Reinforcement learning

•

Deep learning

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
CVLAB  
Event nameEvent placeEvent date
Conference on Neural Information Processing Systems (NeurIPS)

Vancouver, Canada (online)

December 6-12, 2020

RelationURL/DOI

IsSupplementedBy

https://datasets.epfl.ch/disk-data/index.html
Available on Infoscience
November 2, 2020
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/172949
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés