Using wavelet analysis for non-stationary signal de-noising of electro-mechanical system is considered to be the best approach, and wavelet threshold de-noising method is the most simple method that needs the minimum amount of calculation. But this method in selecting threshold functions needs to be improved. Based on different domestic and foreign methods of improving threshold function, propose an improved bivariate threshold function. According to the simulation of non-stationary signal de-noising, the results show that the optimal de-noising results of different signals exist by taking different parameters. Compared with all the de-noising effects, application of the bivariate threshold function considering signal-to-noise ratio and mean square error is superior to the traditional soft and hard threshold functions. At the same time, it can significantly improve the filtering precision, and reserve the main signal details while effectively removing the noise well.