The work is aimed at the investigation of the influence of nonlinear active ions concentration profiles in Yb:YAG laser elements on temperature distribution and wavefront distortions during amplification using sub-kilowatt level diode pumping. A mathematical model is presented for the theoretical study of the amplification process in crystals with cubic crystal system. A detailed comparison of Yb:YAG active elements with the same thickness and absorbed pumping power, but with various concentration profiles of Yb3+, ions is carried out. It is shown that the use of active elements with an increasing dopant concentration in the pump beam direction allows one to optimize the temperature profile inside the active element and, thus, reduce the thermal-induced wavefront distortions of the amplified radiation. Modeling is carried out for the experimentally grown crystal with linear concentration gradient profile. It is shown that the linear doping profile with a gradient of 0.65 at.%/mm allows increasing the small-signal gain up to 10% and decreasing the thermal-induced wavefront distortions by ~15%.