The present disclosure relates to systems, methods, and non-transitory computer readable media for generating counterfactuals low-dimensional balanced nonlinear representations for a matching model. For example, the disclosed systems can utilize an ordinal scatter discrepancy model and a maximum mean discrepancy model to generate low-dimensional balanced nonlinear representations of units. In addition, the disclosed systems can generate counterfactuals based on the low-dimensional balanced nonlinear representations by utilizing a matching model. Further, the disclosed systems can determine an average treatment effect on treated units based on the generated counterfactuals.