KGD-dataset
收藏资源简介:
该数据集是 Amazon Reviews 2023 的一个预处理子集,专为训练和评估 KGD(Knowledge–Geometry Decoupling)与 ManCAR 模型而设计。它覆盖了 8 个产品类别(按字母顺序排列):Arts_Crafts_and_Sewing、Beauty_and_Personal_Care、CDs_and_Vinyl、Cell_Phones_and_Accessories、Office_Products、Software、Toys_and_Games、Video_Games。每个类别都提供了经过严格后处理 pipeline 的数据,包括:按时间排序并采用留一法分割的用户交互序列(训练/验证/测试集)、基于 LightGCN 的协同嵌入(维度256)、基于 Qwen3 的语义嵌入(维度32,从物品标题和层次类别编码而来)、以及用于 ManCAR 的 Manifold-Constrained Swing 图。数据处理流程还包含词汇一致性过滤,确保验证/测试集中不出现训练集未见的物品,从而避免数据泄露。数据集主要用于序列推荐(下一项预测)以及利用协同和语义嵌入的预训练-迁移方法研究。不适用于用户身份识别或高风险决策。使用时需引用原始数据集并说明采用了上述后处理流程。
This dataset is a preprocessed subset of Amazon Reviews 2023, specifically designed for training and evaluating KGD (Knowledge–Geometry Decoupling) and ManCAR models. It covers 8 product categories (in alphabetical order): Arts_Crafts_and_Sewing, Beauty_and_Personal_Care, CDs_and_Vinyl, Cell_Phones_and_Accessories, Office_Products, Software, Toys_and_Games, and Video_Games. Each category provides data processed through a strict post-processing pipeline, including: user interaction sequences sorted by time and split by leave-one-out (training/validation/test sets), collaborative embeddings based on LightGCN (dimension 256), semantic embeddings based on Qwen3 (dimension 32, encoded from item titles and hierarchical categories), and a Manifold-Constrained Swing graph for ManCAR. The data processing pipeline also includes vocabulary consistency filtering to ensure that no unseen items from the training set appear in the validation/test sets, thereby avoiding data leakage. The dataset is mainly used for sequential recommendation (next-item prediction) and research on pretraining-transfer methods utilizing collaborative and semantic embeddings. It is not suitable for user identification or high-risk decision-making. When using, please cite the original dataset and mention that the above post-processing pipeline has been applied.
KGD数据集概述
基本信息
- 许可证: MIT
- 任务类型: 其他(推荐系统相关)
- 标签: 推荐系统、序列推荐、亚马逊评论
数据集简介
该数据集是 Amazon Reviews 2023 的预处理子集,用于训练和评估 KGD(知识-几何解耦)模型,以 ManCAR 作为骨干网络。数据集不仅包含交互数据划分,还包含 KGD 所需的辅助信号:
- 可直接训练的用户交互序列(
train/valid/test) - 用于 BMTP 预训练的协同和语义物品嵌入
- 用于 ManCAR 的流形约束摆动图
覆盖类别(8类)
- Arts_Crafts_and_Sewing
- Beauty_and_Personal_Care
- CDs_and_Vinyl
- Cell_Phones_and_Accessories
- Office_Products
- Software
- Toys_and_Games
- Video_Games
后处理流程(按类别进行)
- 时间排序与留一法划分:交互按时间排序形成用户行为序列,并采用留一法策略划分为训练/验证/测试集。
- 词汇一致性过滤:基于训练集构建物品词汇表,移除验证/测试集中包含未见物品的目标或历史记录,防止训练-测试泄漏。
- 协同嵌入:使用 LightGCN 在仅含训练集边的用户-物品图上训练 256 维嵌入。
- 语义嵌入:使用 Qwen3 文本嵌入模型对物品标题和层级类别编码成 32 维嵌入。
- 摆动图:构建物品-物品摆动相似度图,为 ManCAR 提供流形约束监督。
目录结构(以Software类别为例)
processed_llo_graph/Software/Software.train.csv、Software.valid.csv、Software.test.csv— 用户交互序列Software.item.csv— 物品表(含重映射的物品ID和附加信息)graph/swing.parquet— ManCAR 的流形约束摆动图
graph_emb/Software/graph_emb.csv— 协同(LightGCN)物品嵌入(BMTP)text_emb/Software/text_emb.csv— 语义(Qwen3)物品嵌入(BMTP)
预期用途
该数据集主要用于以下研究:
- 序列推荐 / 下一项预测
- 利用协同和语义物品嵌入的预训练-迁移方法
不适用于用户身份识别或任何高风险决策场景。
许可与引用
- 数据集衍生自 Amazon Reviews 2023,请参考原始数据集页面了解许可、使用条款和归属要求:https://amazon-reviews-2023.github.io/
- 使用该处理数据集时,请引用原始数据集并明确说明使用了上述后处理管线的子集。
- 相关论文:https://arxiv.org/abs/2608.02738
- 相关代码:https://github.com/FuCongResearchSquad/KGD4REC




