During swimming, you open your map, the character changes the position into kneeling. ergo the body is under water so he's "diving" and air goes out.

closing the map -> character swims again and the display is in the same effect like you go with the camera under water, waiting 10-12seconds the display is back normally.