In order to study a new method of image denoising, the auto-encoder module in machine learning is used to optimize the image. By adding noise to the original clarity image, being taken as the input data of the auto-encoder module, and the original image without adding noise as a standard value, the auto-encoder module is trained to improve image denoising ability. Experimental results show that the auto-encoder module has a certain image denoising function after training.