Cross wedge rolling has the advantages of high production efficiency, good product quality, high material utilization, environmental protection, and low cost. It is one of the best processing methods for producing shaft blanks. In this paper, a cross wedge rolling die of TC4 titanium alloy is studied. Based on the Archard wear model, a modified model suitable for cross wedge rolling die wear analysis is derived through finite element simulation. Then, the modified Archard wear model is imported into Deform-3D software for finite element analysis. Orthogonal experimental design is used to combine and analyze different process parameters. Finally, the beetle antennae search (BAS)-genetic algorithm (GA)-back propagation neural network (BPNN) algorithm is used to predict the degree of die wear and to optimize the simulation parameters, which can acquire the process parameters that have the least impact on die wear. The results show that the wear distributions of cross wedge rolling tools is uneven. In general, the most serious areas are basically concentrated in the wedge-shaped inclined plane and rectangular edge lines. The reason is that the tangential force and radial force received by the die are relatively large, which leads to increased wear. Moreover, the temperature change is most severe on the wedge-shaped ridge line. When in contact with the workpiece, the temperature rises sharply, which makes the local temperature rise, the mold hardness decrease, and the wear accelerate. Through response surface method (RSM) analysis, it is concluded that the deformation temperature is the main factor affecting wear depth, followed by the forming angle, and that there is an interaction between the two factors. Finally, the feasibility of the BAS-GA-BP algorithm for optimizing the wear behavior of dies is verified, which provides a new process parameter optimization method for the problem of die wear in the cross wedge rolling process.