Methods and apparatuses for training prompt injection detection model, storage media, and electronic devices

An apparatus for training a prompt injection detection model, comprising:

• at least one processor;
• and
• one or more memories coupled to the at least one processor and storing programming instructions for execution by the at least one processor to perform operations comprising:
• obtaining word feature information corresponding to a prompt training sample, wherein the prompt training sample comprises a normal prompt and a prompt subjected to prompt injection;
• obtaining account feature information based on an account attribute of a user corresponding to the prompt training sample;
• obtaining dialog feature information based on a historical dialog record of the user for a large language model;
• and
• training the prompt injection detection model based on the account feature information, the dialog feature information, and the word feature information, to obtain a trained prompt injection detection model.

A non-transitory, computer-readable medium storing one or more instructions executable by at least one processor to perform operations comprising:

• obtaining word feature information corresponding to a prompt training sample, wherein the prompt training sample comprises a normal prompt and a prompt subjected to prompt injection;
• obtaining account feature information based on an account attribute of a user corresponding to the prompt training sample;
• obtaining dialog feature information based on a historical dialog record of the user for a large language model;
• and
• training the prompt injection detection model based on the account feature information, the dialog feature information, and the word feature information, to obtain a trained prompt injection detection model.