Abstract. By now, a series of advanced wave optical approaches to the processing of radio occultation (RO) observations are widely used. In particular, the canonical transform (CT) method and its further developments need to be mentioned. The latter include the full spectrum inversion (FSI) method, the geometric optical phase matching (PM) method, and the general approach based on the Fourier integral operators (FIOs), also referred to as the CT type 2 (CT2) method. The general idea of these methods is the application of a canonical transform that changes the coordinates in the phase space from time and Doppler frequency to impact parameter and bending angle. For the spherically symmetric atmosphere, the impact parameter, being invariant for each ray, is a unique coordinate of the ray manifold. Therefore, the derivative of the phase of the wave field in the transformed space is directly linked to the bending angle as a single-valued function of the impact parameter. However, in the presence of horizontal gradients, this approach may not work. Here we introduce a further generalization of the CT methods in order to reduce the errors due to horizontal gradients. We describe, in particular, the modified CT2 method, denoted CT2A, which complements the former with one more affine transform: a new coordinate that is a linear combination of the impact parameter and bending angle. The linear combination coefficient is a tunable parameter. We derive the explicit formulas for the CT2A and develop the updated numerical algorithm. For testing the method, we performed statistical analyses based on RO retrievals from data acquired by the Constellation Observing System for Meteorology, Ionosphere, and Climate (COSMIC) and collocated analysis profiles of the European Centre for Medium-Range Weather Forecasts (ECMWF). We demonstrate that it is possible to find a reasonably optimal value of the new tunable CT2A parameter that minimizes the root mean square difference between the RO retrieved and the ECMWF refractivity in the lower troposphere and allows the practical realization of the improved capability to cope with horizontal gradients and serve as the basis of a new quality control procedure.