A computer-implemented method for predicting non-communicable diseases with infectious risk factors using artificial intelligence includes detecting one or more risk factors associated with a non-communicable disease based on a graph associated with person-to-person links, generating a data structure for compactly representing the graph to compute at least one person-to-person distance, and performing a machine learning technique with regularization of the at least one person-to-person distance.