The present disclosure is directed to systems and methods that account for measurement errors acquired from electrodes of a medical device prior to generating an image of the medical device. A correction model is initially generated utilizing a specialized testing catheter that incorporates both electrodes and magnetic sensors. The testing catheter is identically configured to a clinical catheter that has no or fewer magnetic sensors. The testing catheter is used to collect responses (e.g., positions) from the magnetic sensors and electrodes. Electrode positions calculated from the more accurate magnetic sensors are mapped to potentially distorted positions determined from the electrode responses. The correction model (e.g., machine-learning algorithm) is trained using such relationships. Once trained, the correction model is implemented to adjust raw electrode responses prior to generating an image of a medical device. The correction model may adjust raw electrode responses of a clinical usage catheter that lacks magnetic sensors.