The disclosure relates generally to techniques for using information about a user's actual or predicted pupil location for correcting optical distortions that are specific to an optical lens and display assembly through which the user is viewing one or more images. The described techniques may include identifying and mapping optical distortions specific to an optical lens and display assembly, and using such mapped optical distortions to correct images displayed to a wearer or other user receiving images via the assembly, such as based at least in part on pupil location of the wearer or other user. As one example, the one or more optical lens may be mounted inside a head-mounted display (HMD) that also includes a display panel or other image source for an eye of a wearer, and if so one or more pupil tracking mechanisms may be integrated into the HMD.