US 12,039,728 B2
Uncertainty-aware deep reinforcement learning for anatomical landmark detection in medical images
James Browning, Plymouth, MA (US); Li Zhang, Princeton, NJ (US); Benjamin Odry, West New York, NJ (US); Micha Kornreich, New York, NY (US); Jayashri Pawar, Mahwah, NJ (US); Aubrey Chow, Waterford, NY (US); and Richard Herzog, New York, NY (US)
Assigned to Covera Health, New York, NY (US)
Filed by Covera Health, New York, NY (US)
Filed on Feb. 18, 2022, as Appl. No. 17/675,765.
Claims priority of provisional application 63/151,433, filed on Feb. 19, 2021.
Prior Publication US 2022/0270248 A1, Aug. 25, 2022
Int. Cl. G06T 7/00 (2017.01); G06N 3/04 (2023.01); G16H 30/40 (2018.01)
CPC G06T 7/0012 (2013.01) [G06N 3/04 (2013.01); G16H 30/40 (2018.01); G06T 2207/10081 (2013.01); G06T 2207/10088 (2013.01); G06T 2207/20081 (2013.01); G06T 2207/20084 (2013.01); G06T 2207/30004 (2013.01)] 20 Claims
OG exemplary drawing
 
1. A method for anatomical landmark detection, the method comprising:
generating, using a first machine learning sub-network of an anatomical landmark detection agent, one or more image features for a cropped region of interest of a three-dimensional (3D) medical image, wherein the cropped ROI comprises a subset of the 3D medical image centered on a current location of the anatomical landmark detection agent within the 3D medical image;
providing to a second machine learning sub network the one or more image features for use in generating Q-values corresponding to allowable movement directions to be taken by the anatomical landmark detection agent;
generating, using at least a softmax layer of the second machine learning sub-network of the anatomical landmark detection agent for a set of six allowable movement directions associated with movement of the anatomical landmark detection agent within the 3D medical image, six corresponding discrete Q-value distributions wherein the set of six allowable movement directions comprise orthogonal image directions of the 3D medical image and wherein each discrete Q-value distribution:
comprises a predicted value distribution associated with moving the anatomical landmark detection agent from the current location within the 3D medical image, in the respective allowable movement direction;
predicting an anatomical landmark location within the 3D medical image using the plurality of discrete Q-value distributions based on moving the anatomical landmark detection agent within the 3D medical image, wherein in a given state the anatomical landmark detection agent moves in a selected allowable movement direction having a Q-value which is a maximum discrete Q-value distributions expected value over the set of six allowable movement directions; and
determining an uncertainty measure for the predicted anatomical landmark location, wherein the uncertainty measure is determined based on an average full width half maximum (FWHM) value calculated using the respective discrete Q-value distribution having the maximum expected value over the set of six allowable movement directions for the given state.