In Machine Learning conference papers, a common approach is to model relationships between variables using Graph Neural Networks (GNNs). Using GNNs is a powerful and flexible way to go. Maybe you can give it a try!