Embodiments herein disclose a method for controlling a non-linear effect of a power amplifier by an apparatus. The method includes acquiring an input data of the power amplifier of the apparatus and an output data of the power amplifier. Further, the method includes determining an inverse function using a neural network. The inverse function maps normalized output data of the PA to the input data of the PA, where the neural network comprises at least one sub-network for at least one memory tap from a plurality of memory taps in the neural network. Further, the method includes modifying the input data based on the determined inverse function value by dynamically changing a usage of the at least one memory tap from the plurality of memory taps. Further, the method includes compensating the non-linear effect in the output data of the power amplifier.