arXiv:2608.01879cs.LGcs.AI2026-08

用大模型辅助构建表格生成的贝叶斯网络,小样本也能精准模拟真实数据分布。

LAB-Tab: LLM-Augmented Bayesian Network Adaptation for Few-Shot Tabular Generation

论文配图:LAB-Tab: LLM-Augmented Bayesian Network Adaptation for Few-Shot Tabular Generation
图 1 · 摘自论文原文
  • 借助大模型提出目标域缺失的变量关系,扩展源域图结构
  • 通过强化学习优化边权重,在10%数据预算下性能领先
  • 适合数据稀缺但有相关源数据的场景,如医疗、金融建模

当目标域数据稀少时,表格生成可支持分析与决策,但收集完整样本成本高。一种实用但未被充分研究的场景是:仅有少量目标记录,以及来自相关领域的丰富源数据。现有少量样本表格生成方法或直接拟合稀疏的目标统计量,易过拟合偶然模式;或复用源域生成器,可能保留不再成立的依赖关系。为此,我们提出LAB-Tab——一种基于大模型增强的贝叶斯网络(BN)自适应框架,用于源感知的小样本表格生成。LAB-Tab首先从源数据中拟合一个贝叶斯网络,然后利用大模型提出源网络中缺失的目标域变量连接。此步骤将语义和弱统计证据转化为显式结构假设,从而扩展可编辑边的空间。由于所提边可能存在噪声且与已有依赖交互,采用PPO策略通过保持、削弱、加强、翻转、停用等边级操作校准增广后的贝叶斯网络。PPO策略的奖励函数结合了分布对齐性、下游任务效用及目标相关依赖的保留。经调整的贝叶斯网络用于生成目标域表格。在三个美国人口普查(ACS)预测任务构建的六组源-目标分布偏移场景中,LAB-Tab在10%目标数据预算下表现最佳,四次独立场景领先,宏平均得分相比最强基线降低33.8%。同时在宏平均JSD、WAPE和UtilityGap上均取得最优,并保持了良好的特征-标签保真度。

原文摘要 · Abstract (English)

Tabular data generation supports analysis and decision-making when target-domain data are scarce, yet collecting complete target samples is often costly. A practical but underexplored setting provides only a few target records together with richer source data from a related domain. Existing few-shot tabular generators often either fit sparse target statistics directly, which can overfit incidental patterns, or reuse source-domain generators, which may preserve dependencies that no longer hold in the target domain. To address this problem, we propose LAB-Tab, an LLM-augmented Bayesian network (BN) adaptation framework for source-aware few-shot tabular generation. LAB-Tab first fits a BN from source data and then uses an LLM to propose plausible target-domain BN edges that are absent from the source BN graph. This step converts semantic and weak statistical evidence into explicit structural hypotheses, thereby expanding the editable edge space beyond the source-fitted graph. Because the proposed edges may be noisy and interact with existing dependencies, a PPO policy calibrates edges in the augmented BN through edge-level actions, including keep, weaken, strengthen, flip, and deactivate. The PPO policy is trained with a reward that combines distributional alignment, downstream utility, and preservation of target-relevant dependencies. The adapted BN is then sampled to synthesize target-domain tables. Across six source--target distribution-shift scenarios built from three US Census (ACS) prediction tasks, LAB-Tab achieves the best performance at the 10% target-data budget, leads four of the six individual scenarios, and reduces the macro Overall score by 33.8% relative to the strongest baseline. It also obtains the best macro JSD, WAPE, and UtilityGap while maintaining competitive feature--label preservation.

表格生成小样本学习贝叶斯网络大模型应用

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。