The invention relates to optical communication, in particular to compensation of non-linear distortions incurred in high bit-rate optical communication systems. A method and system for compensating self-phase modulation at an optical receiver of an optical transmission system using polarization division multiplexing and a modulation scheme with constant amplitude is proposed. The method comprisesthe step of performing a phase modulation on a received signal, wherein the received signal comprises two signal components associated with two orthogonal polarizations, each component comprising an in-phase sub-component and a quadrature-phase sub-component, thereby spanning a four-dimensional space. The phase modulation is determined by evaluating an error signal which depends on the distance in the four-dimensional space between the received signal after the phase modulation and a four-dimensional sphere defined by target constellation points of the optical transmission system.