Generalizing Spatial Transformers to Projective Geometry with Applications to 2D/3D Registration
Open accessAbstract
Differentiable rendering is a technique to connect 3D scenes with corresponding 2D images. Since it is differentiable, processes during image formation can be learned. Previous approaches to differentiable rendering focus on mesh-based representations of 3D scenes, which is inappropriate for medical applications where volumetric, voxelized models are used to represent anatomy. We propose a novel Projective Spatial Transformer module that generalizes spatial transformers to projective geometry, thus enabling differentiable volume rendering. We demonstrate the usefulness of this architecture on the example of 2D/3D registration between radiographs and CT scans. Specifically, we show that our transformer enables end-to-end learning of an image processing and projection model that approximates an image similarity function that is convex with respect to the pose parameters, and can thus be optimized effectively using conventional gradient descent. To the best of our knowledge, we are the first to describe the spatial transformers in the context of projective transmission imaging, including rendering and pose estimation. We hope that our developments will benefit related 3D research applications. The source code is available at https://github.com/gaocong13/Projective-Spatial-Transformers.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
The multi-scale patch-based feature extraction design learns more robust features than the ResNet block encoders used in the MICCAI published version.
“The multi-scale patch-based feature extraction design learns more robust features than the ResNet block encoders used in the MICCAI published version [31].”
From A Fully Differentiable Framework for 2D/3D Registration and the Projective Spatial Transformers (2024) · cites this paper as [31]
The Methodology section (page unavailable) describes two encoders that produce embedded features, supporting the citation’s identification of an earlier encoder-based design. Our copy does not identify them as ResNet-block encoders or compare them with multi-scale patch-based extraction; the claimed improvement in feature robustness is therefore not established by this source. (LLM, conf=0.78)
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Multimodality image registration by maximization of mutual information
Maes, F., Collignon, A., Vandermeulen, D., Marchal, G., & Suetens, P. (1997) · IEEE Transactions on Medical Imaging · doi:10.1109/42.563664 · Cited by 4,547 · see citing works
Deep learning in medical image registration: a survey
Haskins, G., Kruger, U., & Yan, P. (2020) · Machine Vision and Applications · doi:10.1007/s00138-020-01060-x · Cited by 704 · see citing works
A CNN Regression Approach for Real-Time 2D/3D Registration
Miao, S., Wang, Z. J., & Liao, R. (2016) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2016.2521800 · Cited by 514 · see citing works
3D Pose Regression Using Convolutional Neural Networks
Mahendran, S., Ali, H., & Vidal, R. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) · doi:10.1109/cvprw.2017.73 · Cited by 210 · see citing works
An artificial agent for robust image registration
Liao, R., Miao, S., De Tournemire, P., Grbic, S., Kamen, A., Mansi, T., & Comaniciu, D. (2017) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v31i1.11230 · Cited by 199 · see citing works
Dilated FCN for Multi-Agent 2D/3D Medical Image Registration
Miao, S., Piat, S., Fischer, P., Tuysuzoglu, A., Mewes, P., Mansi, T., & Liao, R. (2018) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v32i1.11576 · Cited by 113 · see citing works
Automatic Annotation of Hip Anatomy in Fluoroscopy for Robust and Efficient 2D/3D Registration
Grupp, R. B., Unberath, M., Gao, C., Hegeman, R. A., Murphy, R. J., Alexander, C. P., Otake, Y., McArthur, B. A., Armand, M., & Taylor, R. H. (2020) · International Journal of Computer Assisted Radiology and Surgery · doi:10.1007/s11548-020-02162-7 · Cited by 81 · cited 1× · see citing works
“Following [2], we formulate an intensity-based 2D/3D registration problem with a pre-operative CT volume V , Digitally Reconstructed Radiograph (DRR) projection operator P , pose parameter θ, a fixed target image If , and a similarity metric loss LS :”
MethodSelf-citationPredicting Slice-to-Volume Transformation in Presence of Arbitrary Subject Motion
Hou, B., Alansary, A., McDonagh, S., Davidson, A., Rutherford, M., Hajnal, J. V., Rueckert, D., Glocker, B., & Kainz, B. (2017) · Lecture Notes in Computer Science · doi:10.1007/978-3-319-66185-8_34 · Cited by 68 · see citing works
Flagged first, then by citations
Research works printed but not linked: 18
- [1]Ferrante E, Oktay O, Glocker B, Milone DH: On the adaptability of unsupervised cnn-based deformable image registration to unseen image domains. In: International Workshop on Machine Learning in Medical Imaging pp. 294– 302. Springer (2018)
- [2]Gibson E, Giganti F, Hu Y, Bonmati E, Bandula S, Gurusamy K, Davidson B, Pereira SP, Clarkson MJ, Barratt DC: Automatic multi-organ segmentation on abdominal ct with dense v-networks. IEEE transactions on medical imaging 37(8), 1822– 1834 (2018) 29994628 PMC6076994
- [4]Hansen N, Müller SD, Koumoutsakos P: Reducing the time complexity of the derandomized evolution strategy with covariance matrix adaptation (cma-es). Evolutionary computation 11(1), 1– 18 (2003) 12804094
- [5]Hartley R, Zisserman A: Multiple view geometry in computer vision. Cambridge university press (2003)
- [7]Henderson P, Ferrari V: Learning single-image 3d reconstruction by generative modelling of shape, pose and shading. International Journal of Computer Vision pp. 1– 20 (2019)
- [9]Jaderberg M, Simonyan K, Zisserman A,: Spatial transformer networks. In: Advances in neural information processing systems. pp. 2017– 2025 (2015)
- [10]Krčah M, Székely G, Blanc R: Fully automatic and fast segmentation of the femur bone from 3d-ct images with no shape prior. In: 2011 IEEE international symposium on biomedical imaging: from nano to macro pp. 2087– 2090. IEEE (2011)
- [11]Krebs J, Mansi T, Delingette H, Zhang L, Ghesu FC, Miao S, Maier AK, Ayache N, Liao R, Kamen A: Robust non-rigid registration through agent-based action learning In: International Conference on Medical Image Computing and Computer-Assisted Intervention. pp. 344– 352. Springer (2017)
- [12]Kuang D, Schmah T: Faim–a convnet method for unsupervised 3d medical image registration. In: International Workshop on Machine Learning in Medical Imaging pp. 646– 654. Springer (2019)
- [14]Liu S, Li T, Chen W, Li H: Soft rasterizer: A differentiable renderer for image-based 3d reasoning. In: Proceedings of the IEEE International Conference on Computer Vision. pp. 7708– 7717 (2019)
- [15]Loper MM, Black MJ: Opendr: An approximate differentiable renderer In: European Conference on Computer Vision. pp. 154– 169. Springer (2014)
- [20]Miolane N, Mathe J, Donnat C, Jorda M, Pennec X: geomstats: a python package for riemannian geometry in machine learning. arXiv preprint arXiv:1805.08308 (2018) · arXiv
- [21]Penney GP, Weese J, Little JA, Desmedt P, Hill DL,: A comparison of similarity measures for use in 2-d-3-d medical image registration. IEEE transactions on medical imaging 17(4), 586– 595 (1998) 9845314
- [22]Roth HR, Lu L, Seff A, Cherry KM, Hoffman J, Wang S, Liu J, Turkbey E, Summers RM: A new 2.5 d representation for lymph node detection using random sets of deep convolutional neural network observations In: International conference on medical image computing and computer-assisted intervention. pp. 520– 527. Springer (2014) PMC4295635 25333158
- [23]Salehi SSM, Khan S, Erdogmus D, Gholipour A: Real-time deep registration with geodesic loss. arXiv preprint arXiv:1803.05982 (2018) · arXiv
- [24]Smith LN: Cyclical learning rates for training neural networks. In: 2017 IEEE Winter Conference on Applications of Computer Vision (WACV) pp. 464– 472. IEEE (2017)
- [25]Würfl T, Hoffmann M, Christlein V, Breininger K, Huang Y, Unberath M, Maier AK: Deep learning computed tomography: Learning projection-domain weights from image domain in limited angle problems. IEEE transactions on medical imaging 37(6), 1454– 1463 (2018) 29870373
- [26]Yan X, Yang J, Yumer E, Guo Y, Lee H: Perspective transformer nets: Learning single-view 3d object reconstruction without 3d supervision. In: Advances in neural information processing systems. pp. 1696– 1704 (2016)
Cited by
Sorted by citations
Unberath, M., Gao, C., Hu, Y., Judish, M., Taylor, R. H., Armand, M., & Grupp, R. (2021) · Frontiers in Robotics and AI · doi:10.3389/frobt.2021.716007 · Cited by 102 · see citing works
“‡ Indeed, among the seven studies that trained deep CNNs on synthetic data and evaluated in a way that allowed for comparisons between synthetic and real data performance, we found quite substantial performance drops [76, 54, 65, 63, 41, 33, 73].”
BackgroundMethodInfluentialSelf-citationA Fully Differentiable Framework for 2D/3D Registration and the Projective Spatial Transformers
Gao, C., Feng, A., Liu, X., Taylor, R. H., Armand, M., & Unberath, M. (2024) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2023.3299588 · Cited by 36 · see citing works
“We also included comparisons with the other representative learning-based 2D/3D registration architectures in the literature : • MICCAI : This is the conference published version of the ProST registration network at MICCAI 2020 [31].”
BackgroundMethodResultInfluentialSelf-citation
- Authors
- Gao, C., Liu, X., Gu, W., Killeen, B., Armand, M., Taylor, R., & Unberath, M.
- Affiliations
- Johns Hopkins University (US)
- Corresponding author
- Gao, Cong
- ORCID
- Venue
- Lecture Notes in Computer Science
- Publisher
- Springer International Publishing
- Published
- 2020
- Volume
- 12263
- Pages
- 329-339
- ISSN
- 0302-9743 (ISSN-L)
- 1611-3349
- PMC
- PMC7602675
- OpenAlex
- W3092242645
- Cited by
- 46
- References
- 8
- Citations by year
- 2026: 3 · 2025: 4 · 2024: 13 · 2023: 9 · 2022: 10 · 2021: 6 · 2020: 1
- FWCI
- 5.78
- Type
- conference paper
- Language
- en
- Open access
- green
- www.ncbi.nlm.nih.gov
- Locations
- Lecture notes in computer science (publishedVersion)
- PubMed (publishedVersion)
- PubMed Central (acceptedVersion)
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Funders
- National Institute of Biomedical Imaging and Bioengineering (R21EB020113, R01EB023939, R21EB028505)
- Topics
- Physical Sciences › Engineering › Computational Mechanics › 3D Shape Modeling and Analysis
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Medical Image Segmentation Techniques
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Vision and Imaging
- Keywords
- spatial transformer network, projective geometry, differentiable rendering, 2D-3D registration, pose estimation, volumetric medical images
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review