Imaging 3D polarization dynamics via deep learning 4D-STEM