The invention discloses a molecular optimization method for
protein pocket three-dimensional structure
perception based on a
diffusion model, and relates to the field of molecular optimization. The method comprises the following steps: enabling a
diffusion model constructed based on an EGNN to perform forward
diffusion on a
supervised training set so as to learn correct data distribution of
atomic coordinates, atomic types and key types in optimized molecules, and obtaining an initial model; constructing a
protein-ligand graph on the basis of a k-nearest
neighbor algorithm, and constructing a full-connection ligand graph on the basis of bond types between atoms; carrying out
reverse diffusion on the initial model, carrying out
information transmission by utilizing a
protein-ligand graph and a full-connection ligand graph, and carrying out iterative updating to obtain a molecular optimization model; and carrying out fragment connection, skeleton modification or skeleton transition by utilizing the molecular optimization model. Compared with the prior art, the method is applicable to fragment connection, skeleton modification and skeleton transition, so that the
small molecule optimization process is more convenient, the three-dimensional protein pocket sensing effect is achieved, and the problem that more invalid molecules and non-
real structure compounds are generated by a conventional model is solved.