46 citations · 156 across the 29 of their papers we have counts for
30 papers
AgentDR: Dynamic Recommendation with Implicit Item-Item Relations via LLM-based Agents
Mingdai Yang, Nurendra Choudhary, Jiangshu Du +4
Recent agent-based recommendation frameworks aim to simulate user behaviors by incorporating memory mechanisms and prompting strategies, but they struggle with hallucinating non-ex…
Optimas: Optimizing Compound AI Systems with Globally Aligned Local Rewards
Shirley Wu, Parth Sarthi, Shiyu Zhao +10
Compound AI systems integrating multiple components, such as Large Language Models, specialized tools, and traditional machine learning models, are increasingly deployed to solve c…
GT2Vec: Large Language Models as Multi-Modal Encoders for Text and Graph-Structured Data
Jiacheng Lin, Kun Qian, Haoyu Han +9
Graph-structured information offers rich contextual information that can enhance language models by providing structured relationships and hierarchies, leading to more expressive e…
Covering a Graph with Dense Subgraph Families, via Triangle-Rich Sets
Sabyasachi Basu, Daniel Paul-Pena, Kun Qian +3
Graphs are a fundamental data structure used to represent relationships in domains as diverse as the social sciences, bioinformatics, cybersecurity, the Internet, and more. One of…
All Against Some: Efficient Integration of Large Language Models for Message Passing in Graph Neural Networks
Ajay Jaiswal, Nurendra Choudhary, Ravinarayana Adkathimar +6
Graph Neural Networks (GNNs) have attracted immense attention in the past decade due to their numerous real-world applications built around graph-structured data. On the other hand…
AvaTaR: Optimizing LLM Agents for Tool Usage via Contrastive Reasoning
Shirley Wu, Shiyu Zhao, Qian Huang +7
Large language model (LLM) agents have demonstrated impressive capabilities in utilizing external tools and knowledge to boost accuracy and reduce hallucinations. However, developi…
STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Bases
Shirley Wu, Shiyu Zhao, Michihiro Yasunaga +7
Answering real-world complex queries, such as complex product search, often requires accurate retrieval from semi-structured knowledge bases that involve blend of unstructured (e.g…
An Interpretable Ensemble of Graph and Language Models for Improving Search Relevance in E-Commerce
Nurendra Choudhary, Edward W Huang, Karthik Subbian +1
The problem of search relevance in the E-commerce domain is a challenging one since it involves understanding the intent of a user's short nuanced query and matching it with the ap…
Graph Coarsening via Convolution Matching for Scalable Graph Neural Network Training
Charles Dickens, Eddie Huang, Aishwarya Reganti +3
Graph summarization as a preprocessing step is an effective and complementary technique for scalable graph neural network (GNN) training. In this work, we propose the Coarsening Vi…
ForeSeer: Product Aspect Forecasting Using Temporal Graph Embedding
Zixuan Liu, Gaurush Hiranandani, Kun Qian +5
Developing text mining approaches to mine aspects from customer reviews has been well-studied due to its importance in understanding customer needs and product attributes. In contr…
Communication-Free Distributed GNN Training with Vertex Cut
Kaidi Cao, Rui Deng, Shirley Wu +3
Training Graph Neural Networks (GNNs) on real-world graphs consisting of billions of nodes and edges is quite challenging, primarily due to the substantial memory needed to store t…
Amazon-M2: A Multilingual Multi-locale Shopping Session Dataset for Recommendation and Text Generation
Wei Jin, Haitao Mao, Zheng Li +17
Modeling customer shopping intentions is a crucial task for e-commerce, as it directly impacts user experience and engagement. Thus, accurately understanding customer preferences i…
Simplifying Distributed Neural Network Training on Massive Graphs: Randomized Partitions Improve Model Aggregation
Jiong Zhu, Aishwarya Reganti, Edward Huang +4
Distributed training of GNNs enables learning on massive graphs (e.g., social and e-commerce networks) that exceed the storage and computational capacity of a single machine. To re…
You Only Transfer What You Share: Intersection-Induced Graph Transfer Learning for Link Prediction
Wenqing Zheng, Edward W Huang, Nikhil Rao +2
Link prediction is central to many real-world applications, but its performance may be hampered when the graph of interest is sparse. To alleviate issues caused by sparsity, we inv…
Search Behavior Prediction: A Hypergraph Perspective
Yan Han, Edward W Huang, Wenqing Zheng +3
Although the bipartite shopping graphs are straightforward to model search behavior, they suffer from two challenges: 1) The majority of items are sporadically searched and hence h…
TuneUp: A Simple Improved Training Strategy for Graph Neural Networks
Weihua Hu, Kaidi Cao, Kexin Huang +4
Despite recent advances in Graph Neural Networks (GNNs), their training strategies remain largely under-explored. The conventional training strategy learns over all nodes in the or…
Text Enriched Sparse Hyperbolic Graph Convolutional Networks
Nurendra Choudhary, Nikhil Rao, Karthik Subbian +1
Heterogeneous networks, which connect informative nodes containing text with different edge types, are routinely used to store and process information in various real-world applica…
Shopping Queries Dataset: A Large-Scale ESCI Benchmark for Improving Product Search
Chandan K. Reddy, Lluís Màrquez, Fran Valero +6
Improving the quality of search results can significantly enhance users experience and engagement with search engines. In spite of several recent advancements in the fields of mach…
Learning Backward Compatible Embeddings
Weihua Hu, Rajas Bansal, Kaidi Cao +3
Embeddings, low-dimensional vector representation of objects, are fundamental in building modern machine learning systems. In industrial settings, there is usually an embedding tea…
Multilingual Knowledge Graph Completion with Self-Supervised Adaptive Graph Alignment
Zijie Huang, Zheng Li, Haoming Jiang +6
Predicting missing facts in a knowledge graph (KG) is crucial as modern KGs are far from complete. Due to labor-intensive human labeling, this phenomenon deteriorates when handling…
Task-Agnostic Graph Explanations
Yaochen Xie, Sumeet Katariya, Xianfeng Tang +4
Graph Neural Networks (GNNs) have emerged as powerful tools to encode graph-structured data. Due to their broad applications, there is an increasing need to develop tools to explai…
Cold Brew: Distilling Graph Node Representations with Incomplete or Missing Neighborhoods
Wenqing Zheng, Edward W Huang, Nikhil Rao +3
Graph Neural Networks (GNNs) have achieved state-of-the-art performance in node classification, regression, and recommendation tasks. GNNs work well when rich and high-quality conn…
Probabilistic Entity Representation Model for Reasoning over Knowledge Graphs
Nurendra Choudhary, Nikhil Rao, Sumeet Katariya +2
Logical reasoning over Knowledge Graphs (KGs) is a fundamental technique that can provide efficient querying mechanism over large and incomplete databases. Current approaches emplo…
Scalable Feature Selection for (Multitask) Gradient Boosted Trees
Cuize Han, Nikhil Rao, Daria Sorokina +1
Gradient Boosted Decision Trees (GBDTs) are widely used for building ranking and relevance models in search and recommendation. Considerations such as latency and interpretability…