Search papers, labs, and topics across Lattice.
This study introduces a prior-guided hybrid concept bottleneck model (CBM) that enhances the interpretability of cancer imaging diagnostics while significantly reducing the need for extensive instance-level annotations. By integrating limited concept annotations with class-conditional concept distribution matching and prior initialization, the model demonstrates improved mean concept AUC in scenarios with as little as 10% annotation, outperforming standard CBMs. The findings indicate that structured priors can effectively alleviate the annotation burden while maintaining diagnostic accuracy comparable to black-box models.
A hybrid concept bottleneck model boosts interpretability in cancer imaging diagnostics with only 10% annotation, achieving a mean concept AUC increase from 0.619 to 0.741 for mammographic masses.
Concept bottleneck models (CBMs) can improve the transparency of cancer image diagnostic prediction by expressing predictions through radiological concepts. However, their dependence on instance-level concept annotations limits practical applicability. We propose a prior-guided hybrid CBM that integrates limited concept annotations, class-conditional concept distribution matching on unannotated patients, and prior initialization of the concept-to-diagnosis head. We evaluate the method on CBIS-DDSM mammographic masses and calcifications and LIDC-IDRI pulmonary nodules across 0-100% concept annotation. In the clinically relevant 0-20% annotation regime, the hybrid CBM consistently improves mean concept AUC over a matched standard CBM, while maintaining diagnostic performance close to black-box models. At 10% annotation specifically, concept AUC increases from 0.619 to 0.741 for masses, from 0.650 to 0.787 for calcifications, and from 0.597 to 0.642 for pulmonary nodules. Ablation experiments identify prior initialization as the main component contributing to improved concept detection, likely by stabilizing the concept-to-diagnosis head. Zero-shot VLMs remain insufficient for reliable fine-grained tumor-level concept prediction. These findings suggest that structured priors can substantially reduce the annotation burden of interpretable cancer imaging models.
