Methods and apparatuses (including software) for optimizing dental treatment plans, including for optimizing the treatment plans using dental aligners. These methods and apparatuses may optimize a treatment plan by estimating a difference between a target set of tooth position and a predicted set of tooth positions using the treatment plan using a prediction network trained to use multiple translational and rotational directions for individual teeth as well as reaction forces on the individual teeth based on adjacent teeth.