Generalizable stereo depth estimation with masked image modelling
Author(s)
Tukra, Samyakh
Xu, Haozheng
Xu, Chi
Giannarou, Stamatia
Type
Journal Article
Abstract
Generalizable and accurate stereo depth estimation is vital for 3D reconstruction, especially in surgery. Supervised learning methods obtain best performance however, limited ground truth data for surgical scenes limits generalizability. Self-supervised methods don't need ground truth, but suffer from scale ambiguity and incorrect disparity prediction due to inconsistency of photometric loss. This work proposes a two-phase training procedure that is generalizable and retains the high performance of supervised methods. It entails: (1) performing self-supervised representation learning of left and right views via masked image modelling (MIM) to learn generalizable semantic stereo features (2) utilizing the MIM pre-trained model to learn robust depth representation via supervised learning for disparity estimation on synthetic data only. To improve stereo representations learnt via MIM, perceptual loss terms are introduced, which improve the model's stereo representations learnt by explicitly encouraging the learning of higher scene-level features. Qualitative and quantitative performance evaluation on surgical and natural scenes shows that the approach achieves sub-millimetre accuracy and lowest errors respectively, setting a new state-of-the-art. Despite not training on surgical nor natural scene data for disparity estimation.
Date Issued
2024-04
Date Acceptance
2023-12-04
Citation
Healthcare Technology Letters, 2024, 11 (2-3), pp.108-116
ISSN
2053-3713
Publisher
Wiley
Start Page
108
End Page
116
Journal / Book Title
Healthcare Technology Letters
Volume
11
Issue
2-3
Copyright Statement
© 2023 The Authors. Healthcare Technology Letters published by John Wiley & Sons Ltd on behalf of The Institution of Engineering and Technology. This is an open access article under the terms of the Creative Commons Attribution License, which permits use, distribution and reproduction in any medium, provided the original work is
properly cited.
properly cited.
License URL
Identifier
http://dx.doi.org/10.1049/htl2.12067
Publication Status
Published
Date Publish Online
2023-12-23