Multi-Quantifying Maxillofacial Traits via a Demographic Parity-Based AI Model
Mengru Shi,Zhuohong Gong,Peisheng Zeng,Dawei Xiang,Gengbin Cai,Hengyi Liu,Shijie Chen,Runheng Liu,Zhuofan Chen,Xinchun Zhang,Zetao Chen
DOI: https://doi.org/10.34133/bmef.0054
2024-08-13
Abstract:Objective and Impact Statement: The multi-quantification of the distinct individualized maxillofacial traits, that is, quantifying multiple indices, is vital for diagnosis, decision-making, and prognosis of the maxillofacial surgery. Introduction: While the discrete and demographically disproportionate distributions of the multiple indices restrict the generalization ability of artificial intelligence (AI)-based automatic analysis, this study presents a demographic-parity strategy for AI-based multi-quantification. Methods: In the aesthetic-concerning maxillary alveolar basal bone, which requires quantifying a total of 9 indices from length and width dimensional, this study collected a total of 4,000 cone-beam computed tomography (CBCT) sagittal images, and developed a deep learning model composed of a backbone and multiple regression heads with fully shared parameters to intelligently predict these quantitative metrics. Through auditing of the primary generalization result, the sensitive attribute was identified and the dataset was subdivided to train new submodels. Then, submodels trained from respective subsets were ensembled for final generalization. Results: The primary generalization result showed that the AI model underperformed in quantifying major basal bone indices. The sex factor was proved to be the sensitive attribute. The final model was ensembled by the male and female submodels, which yielded equal performance between genders, low error, high consistency, satisfying correlation coefficient, and highly focused attention. The ensemble model exhibited high similarity to clinicians with minor processing time. Conclusion: This work validates that the demographic parity strategy enables the AI algorithm with greater model generalization ability, even for the highly variable traits, which benefits for the appearance-concerning maxillofacial surgery.