BEA-CACE: branch-endpoint-aware double-DQN for coronary artery centerline extraction in CT angiography images.
Authors
Affiliations (6)
Affiliations (6)
- Faculty of Computing, Harbin Institute of Technology, Harbin, 150001, China.
- School of Computer Science and Technology, Harbin Institute of Technology, Shenzhen, 518055, China.
- Department of Radiology, Fourth Affiliated Hospital of Harbin Medical University, Harbin, 150001, China.
- College of Computer and Control Engineering, Northeast Forestry University, Harbin, 150040, China.
- Department of Cardiology, Fourth Affiliated Hospital of Harbin Medical University, Harbin, 150001, China.
- Faculty of Computing, Harbin Institute of Technology, Harbin, 150001, China. [email protected].
Abstract
In order to automate the centerline extraction of the coronary tree, three challenges must be addressed: tracking branches automatically, passing through plaques successfully, and detecting endpoints accurately. This study aims to develop a method to solve the three challenges. We propose a branch-endpoint-aware coronary centerline extraction framework. The framework consists of a deep reinforcement learning-based tracker and a 3D dilated CNN-based detector. The tracker is designed to predict the actions of an agent with the objective of tracking the centerline. The detector identifies bifurcation points and endpoints, assisting the tracker in tracking branches and terminating the tracking process automatically. The detector can also estimate the radius values of the coronary artery. The method achieves the state-of-the-art performance in both the centerline extraction and radius estimate. Furthermore, the method necessitates minimal user interaction to extract a coronary tree, a feature that surpasses other interactive methods. The method can track branches automatically, pass through plaques successfully and detect endpoints accurately. Compared with other interactive methods that require multiple seeds, our method only needs one seed to extract the entire coronary tree.