diff --git a/packages/service/core/dataset/data/schema.ts b/packages/service/core/dataset/data/schema.ts
index d291f3cec..7100d06ce 100644
--- a/packages/service/core/dataset/data/schema.ts
+++ b/packages/service/core/dataset/data/schema.ts
@@ -107,6 +107,16 @@ try {
 
   // Cron clear invalid data
   DatasetDataSchema.index({ updateTime: 1 });
+
+  // fileId index
+  DatasetDataSchema.index(
+    { fileId: 1 },
+    { unique: true, partialFilterExpression: { fileId: { $exists: true } } }
+  );
+  DatasetDataSchema.index(
+    { imageId: 1 },
+    { unique: true, partialFilterExpression: { imageId: { $exists: true } } }
+  );
 } catch (error) {
   console.log(error);
 }