The artificial intelligence algorithm of the AI controller was researched, based on the DH-TSD thermally stimulated discharge apparatus developed by ourselves, and a comparison was made between the temperature rising curves of AI control and step control, the results show that the use of artificial intelligence algorithm can get a better linear for the temperature rising curve, which is conducive to raising the accuracy of the thermally stimulated discharge experiment. At last we demonstrated the conclusion by TSD experiment on the self-developed apparatus.