Machine Learning for Neonatal Mortality Risk Assessment: A Case Study Using Public Health Data from São Paulo

Abstract

AbstractInfant mortality is a reflection of a complex combination of biological, socioeconomic and health care factors that require various data sources for a thorough analysis. Consequently, the use of specialized tools and techniques to deal with a large volume of data is extremely helpful. Machine learning has been applied to solve problems from many domains and presents great potential for the proposed problem, which would be an innovation in Brazilian reality. In this paper, an innovative method is proposed to perform a neonatal death risk assessment using computer vision techniques. Using mother, pregnancy care and child at birth features, from a dataset containing neonatal samples from São Paulo city public health data, the proposed method encodes images features and uses a custom convolutional neural network architecture to classification. Experiments show that the method is able to detect death samples with accuracy of 90.61%.