collaborators

5 papers

cs.AI2026

Beyond Predefined Schemas: TRACE-KG for Context-Enriched Knowledge Graph Generation

Mohammad Sadeq Abolhasani, Yang Ba, Yixuan He +1

Knowledge graph generation typically relies either on predefined ontologies or on schema-free extraction. Ontology-driven pipelines enforce consistent typing but require costly sch…

cs.AI2026

MemArchitect: A Policy Driven Memory Governance Layer

Lingavasan Suresh Kumar, Yang Ba, Rong Pan

Persistent Large Language Model (LLM) agents expose a critical governance gap in memory management. Standard Retrieval-Augmented Generation (RAG) frameworks treat memory as passive…

cs.AI2026

Measuring Dataset Diversity from a Geometric Perspective

Yang Ba, Mohammad Sadeq Abolhasani, Michelle V Mancenido +1

Diversity can be broadly defined as the presence of meaningful variation across elements, which can be viewed from multiple perspectives, including statistical variation and geomet…

cs.LG2025

Predict Training Data Quality via Its Geometry in Metric Space

Yang Ba, Mohammad Sadeq Abolhasani, Rong Pan

High-quality training data is the foundation of machine learning and artificial intelligence, shaping how models learn and perform. Although much is known about what types of data…

cs.LG2025

Data Diversity as Implicit Regularization: How Does Diversity Shape the Weight Space of Deep Neural Networks?

Yang Ba, Michelle V. Mancenido, Rong Pan

Data augmentation that introduces diversity into the input data has long been used in training deep learning models. It has demonstrated benefits in improving robustness and genera…