...mine did the same thing and it does come down to imu calibration,compass calibration,remote stick calibration and even vision sensor calibration...which can be done on DJI Assistant 2.
Doing these calibrations and keeping my Air away from anything magnetic and any objects that can be subject...