Search papers, labs, and topics across Lattice.
This paper introduces LAB-Tab, a novel framework that leverages LLMs to enhance Bayesian network adaptation for few-shot tabular data generation in scenarios with limited target-domain records. By fitting a Bayesian network from source data and using an LLM to suggest plausible target-domain edges, LAB-Tab effectively expands the structural hypotheses beyond the source data, mitigating overfitting and preserving relevant dependencies. The framework outperforms existing methods across multiple distribution-shift scenarios, achieving a 33.8% reduction in macro Overall score compared to the strongest baseline while maintaining competitive feature-label preservation.
LAB-Tab reduces overfitting in few-shot tabular generation by intelligently expanding Bayesian network structures with LLM-driven insights, leading to significant performance gains.
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.
