Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Ni Lao

dblp:82/283 · DBLP profile ↗
← Back
34ranked-venue papers
7as first author
13since 2021 · last 2025
0000-0002-4034-7784ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 24 · 6 first-author · 8 since 2021Databases, data management, data science and information retrieval · 15 · 1 first-author · 8 since 2021Systems, architecture and hardware · 3 · 1 first-authorApplied, interdisciplinary, general and emerging computing · 3 · 3 since 2021Security and privacy · 1Graphics, computer vision, multimedia, augmented reality and games · 1

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
12 papers
Representation and self-supervised learning · 21% Trustworthy machine learning · 14% Language models and text generation · 12%
Databases, data mining, and information retrieval
11 papers
Information retrieval · 51% Data mining · 40% Knowledge graphs · 7%

Topics — the 30 heaviest of 64, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Representation and self-supervised learning
spatial representation learning
1.222024
TorchSpatial: A Location Encoding Framework and Benchmark for Spatial Representation Learning · NeurIPS 2024
Multi-Scale Representation Learning for Spatial Feature Distributions using Grid Cells · ICLR 2020
Machine learning › Trustworthy machine learning
fairness
0.812024
TorchSpatial: A Location Encoding Framework and Benchmark for Spatial Representation Learning · NeurIPS 2024
Machine learning › Trustworthy machine learning › dataset bias
geographic bias
0.812024
TorchSpatial: A Location Encoding Framework and Benchmark for Spatial Representation Learning · NeurIPS 2024
Machine learning › Deep learning architectures and training
positional encoding
0.812024
TorchSpatial: A Location Encoding Framework and Benchmark for Spatial Representation Learning · NeurIPS 2024
Data mining
clustering
0.812024
MC-GTA: Metric-Constrained Model-Based Clustering using Goodness-of-fit Tests with Autocorrelations · ICML 2024
Information retrieval
image retrieval
0.812024
Img2Loc: Revisiting Image Geolocalization using Multi-modality Foundation Models and Image-based Retrieval-Augmented Generation · SIGIR 2024
Data mining › clustering
model-based clustering
0.812024
MC-GTA: Metric-Constrained Model-Based Clustering using Goodness-of-fit Tests with Autocorrelations · ICML 2024
Information retrieval
retrieval-augmented generation
0.812024
Img2Loc: Revisiting Image Geolocalization using Multi-modality Foundation Models and Image-based Retrieval-Augmented Generation · SIGIR 2024
Data mining › clustering › spatial clustering
spatio-temporal clustering
0.812024
MC-GTA: Metric-Constrained Model-Based Clustering using Goodness-of-fit Tests with Autocorrelations · ICML 2024
Information retrieval › multimedia analysis and retrieval
visual geolocalization
0.812024
Img2Loc: Revisiting Image Geolocalization using Multi-modality Foundation Models and Image-based Retrieval-Augmented Generation · SIGIR 2024
Machine learning › Representation and self-supervised learning
contrastive learning
0.712023
CSP: Self-Supervised Contrastive Spatial Pre-Training for Geospatial-Visual Representations · ICML 2023
Natural language and speech › Language models and text generation › trustworthy language model › large language model reliability › factuality
factual consistency
0.712023
RARR: Researching and Revising What Language Models Say, Using Language Models · ACL (1) 2023
Natural language and speech › Language models and text generation
text generation
0.712023
RARR: Researching and Revising What Language Models Say, Using Language Models · ACL (1) 2023
Information retrieval
evidence retrieval
0.712023
RARR: Researching and Revising What Language Models Say, Using Language Models · ACL (1) 2023
Machine learning › Representation and self-supervised learning › computational neuroscience › neural coding
grid cell
0.412020
Multi-Scale Representation Learning for Spatial Feature Distributions using Grid Cells · ICLR 2020
Machine learning › Reinforcement learning › policy optimization
policy gradient
0.312018
Memory Augmented Policy Optimization for Program Synthesis and Semantic Parsing · NeurIPS 2018
Machine learning › Optimization for machine learning
variance reduction
0.312018
Memory Augmented Policy Optimization for Program Synthesis and Semantic Parsing · NeurIPS 2018
Knowledge, reasoning and agents › Knowledge representation and reasoning
relational learning
0.322015
Learning Relational Features with Backward Random Walks · ACL (1) 2015
Efficient Relational Learning with Hidden Variable Detection · NIPS 2010
Knowledge, reasoning and agents › Knowledge representation and reasoning › query answering
knowledge base querying
0.312017
Neural Symbolic Machines: Learning Semantic Parsers on Freebase with Weak Supervision · ACL (1) 2017
Natural language and speech › Information extraction and text analysis
semantic parsing
0.312017
Neural Symbolic Machines: Learning Semantic Parsers on Freebase with Weak Supervision · ACL (1) 2017
Natural language and speech › Information extraction and text analysis › semantic parsing
weakly supervised semantic parsing
0.312017
Neural Symbolic Machines: Learning Semantic Parsers on Freebase with Weak Supervision · ACL (1) 2017
Knowledge, reasoning and agents › Knowledge representation and reasoning › knowledge acquisition
never-ending learning
0.212015
Never-Ending Learning · AAAI 2015
Knowledge, reasoning and agents › Knowledge representation and reasoning › ontology › ontology evolution
ontology extension
0.212015
Never-Ending Learning · AAAI 2015
Natural language and speech › Information extraction and text analysis
web information extraction
0.212015
Never-Ending Learning · AAAI 2015
Computer vision › Image recognition and object detection
image classification
0.212023
CSP: Self-Supervised Contrastive Spatial Pre-Training for Geospatial-Visual Representations · ICML 2023
Computer vision › 3D vision
remote sensing
0.212023
CSP: Self-Supervised Contrastive Spatial Pre-Training for Geospatial-Visual Representations · ICML 2023
Knowledge graphs › knowledge graph construction
knowledge extraction
0.212014
Knowledge vault: a web-scale approach to probabilistic knowledge fusion · KDD 2014
Knowledge graphs
knowledge graph construction
0.212014
Knowledge vault: a web-scale approach to probabilistic knowledge fusion · KDD 2014
Natural language and speech › Information extraction and text analysis
open information extraction
0.112012
Reading The Web with Learned Syntactic-Semantic Inference Rules · EMNLP-CoNLL 2012
Machine learning › Deep learning architectures and training
multi-scale representation
0.112020
Multi-Scale Representation Learning for Spatial Feature Distributions using Grid Cells · ICLR 2020

Methods — techniques the papers use, named apart from their topics

retrieval-augmented generation · 2.1language model revision · 1.3wasserstein distance · 0.8semivariogram · 0.8self-supervised learning · 0.8multimodal large language model prompting · 0.8goodness-of-fit test · 0.8diffusion transformer · 0.8CLIP representation · 0.8systematic exploration · 0.7memory augmented policy optimization · 0.7dual encoder · 0.7distributed sampling · 0.7contrastive learning · 0.7grid cell modeling · 0.4coordinate descent · 0.3supervised machine learning · 0.2probabilistic inference · 0.2
YearPublicationVenuePosition
2025 LocDiff: Identifying Locations on Earth by Diffusing in the Hilbert Space
abstract
Image geolocalization is a fundamental yet challenging task, aiming at inferring the geolocation on Earth where an image is taken. State-of-the-art methods employ either grid-based classification or gallery-based image-location retrieval, whose spatial generalizability significantly suffers if the spatial distribution of test images does not align with the choices of grids and galleries. Recently emerging generative approaches, while getting rid of grids and galleries, use raw geographical coordinates and suffer quality losses due to their lack of multi-scale information. To address these limitations, we propose a multi-scale latent diffusion model called LocDiff for image geolocalization. We developed a novel positional encoding-decoding framework called Spherical Harmonics Dirac Delta (SHDD) Representations, which encodes points on a spherical surface (e.g., geolocations on Earth) into a Hilbert space of Spherical Harmonics coefficients and decodes points (geolocations) by mode-seeking on spherical probability distributions. We also propose a novel SirenNet-based architecture (CS-UNet) to learn an image-based conditional backward process in the latent SHDD space by minimizing a latent KL-divergence loss. To the best of our knowledge, LocDiff is the first image geolocalization model that performs latent diffusion in a multi-scale location encoding space and generates geolocations under the guidance of images. Experimental results show that LocDiff can outperform all state-of-the-art grid-based, retrieval-based, and diffusion-based baselines across 5 challenging global-scale image geolocalization datasets, and demonstrates significantly stronger generalizability to unseen geolocations.
Zeping Liu, Jielu Zhang, Zhongliang Zhou, Nemin Wu, Lan Mu, Yiqun Xie, Ni Lao, Gengchen Mai
NeurIPS10
2025 GeoFM: how will geo-foundation models reshape spatial data science and GeoAI?
abstract
The emerging field of geo-foundation models (GeoFM) has the potential to reshape GeoAI and spatial data science research, education, and practice. In this work, we motivate and define the term and put it into its historic context within GeoAI and spatial data science more broadly. Next, we review core datasets, models, and benchmarks. Based on this overview of the state-of-the-art, we introduce key research challenges for future GeoFM research, such as GeoAI scaling laws, geo-alignment of AI, truly multimodal GeoFM, and so on. Finally, we discuss potential risks of GeoFM research and outline the road ahead with a specific focus on the increasing role of international large-scale collaborations and the future of GeoAI and spatial data science education.
Krzysztof Janowicz, Gengchen Mai, Weiming Huang 0001, Rui Zhu 0008, Ni Lao, Ling Cai 0002
Int. J. Geogr. Inf. Sci.5
2024 SRL: Towards a General-Purpose Framework for Spatial Representation Learning
abstract
Representation learning (RL) techniques are widely adopted in areas such as natural language processing and computer vision, with prominent examples such as attention and ConvNet architectures. In comparison, many GeoAI works still rely on feature engineering or data conversion to represent spatial data (e.g., points, polylines, polygons, 3D building models, etc.) as features in formats that are easier for neural networks to handle. The neural network architectures remain unchanged, and the need for feature engineering has become a bottleneck for applying deep learning to new tasks in the age of big data. In this paper, we advocate the idea of developing learnable spatial representation modules, which not only enable spatial reasoning but also enable neural nets to directly consume (i.e., encoding) or generate (i.e., decoding) spatial data. We propose Spatial Representation Learning (SRL), a new general-purpose representation learning framework for spatial reasoning. We discuss the key challenges of spatial representation learning including multi-scale RL, continuous RL, shape-centric RL, noise-robust RL, heterogeneity-aware RL, and fairness-aware RL. We also discuss the critical role and potential of SRL in various geospatial subdomains and how this technique can lead to a new generation of GeoAI.
Gengchen Mai, Xiaobai Angela Yao, Yiqun Xie, Jinmeng Rao, Hao Li 0019, Qing Zhu 0011, Ni Lao
SIGSPATIAL/GIS8
2024 MC-GTA: Metric-Constrained Model-Based Clustering using Goodness-of-fit Tests with Autocorrelations
abstract
A wide range of (multivariate) temporal (1D) and spatial (2D) data analysis tasks, such as grouping vehicle sensor trajectories, can be formulated as clustering with given metric constraints. Existing metric-constrained clustering algorithms overlook the rich correlation between feature similarity and metric distance, i.e., metric autocorrelation. The model-based variations of these clustering algorithms (e.g. TICC and STICC) achieve SOTA performance, yet suffer from computational instability and complexity by using a metric-constrained Expectation-Maximization procedure. In order to address these two problems, we propose a novel clustering algorithm, MC-GTA (Model-based Clustering via Goodness-of-fit Tests with Autocorrelations). Its objective is only composed of pairwise weighted sums of feature similarity terms (square Wasserstein-2 distance) and metric autocorrelation terms (a novel multivariate generalization of classic semivariogram). We show that MC-GTA is effectively minimizing the total hinge loss for intra-cluster observation pairs not passing goodness-of-fit tests, i.e., statistically not originating from the same distribution. Experiments on 1D/2D synthetic and real-world datasets demonstrate that MC-GTA successfully incorporates metric autocorrelation. It outperforms strong baselines by large margins (up to 14.3% in ARI and 32.1% in NMI) with faster and stabler optimization ($>$10x speedup).
Gengchen Mai, Krzysztof Janowicz, Ni Lao
ICML4
2024 TorchSpatial: A Location Encoding Framework and Benchmark for Spatial Representation Learning
abstract
Spatial representation learning (SRL) aims at learning general-purpose neural network representations from various types of spatial data (e.g., points, polylines, polygons, networks, images, etc.) in their native formats. Learning good spatial representations is a fundamental problem for various downstream applications such as species distribution modeling, weather forecasting, trajectory generation, geographic question answering, etc. Even though SRL has become the foundation of almost all geospatial artificial intelligence (GeoAI) research, we have not yet seen significant efforts to develop an extensive deep learning framework and benchmark to support SRL model development and evaluation. To fill this gap, we propose TorchSpatial, a learning framework and benchmark for location (point) encoding,which is one of the most fundamental data types of spatial representation learning. TorchSpatial contains three key components: 1) a unified location encoding framework that consolidates 15 commonly recognized location encoders, ensuring scalability and reproducibility of the implementations; 2) the LocBench benchmark tasks encompassing 7 geo-aware image classification and 10 geo-aware imageregression datasets; 3) a comprehensive suite of evaluation metrics to quantify geo-aware models’ overall performance as well as their geographic bias, with a novel Geo-Bias Score metric. Finally, we provide a detailed analysis and insights into the model performance and geographic bias of different location encoders. We believe TorchSpatial will foster future advancement of spatial representationlearning and spatial fairness in GeoAI research. The TorchSpatial model framework and LocBench benchmark are available at https://github.com/seai-lab/TorchSpatial, and the Geo-Bias Score evaluation framework is available at https://github.com/seai-lab/PyGBS.
Nemin Wu, Zeping Liu, Yanlin Qi, Jielu Zhang, Joshua Ni, Xiaobai Angela Yao, Lan Mu, Stefano Ermon, Tanuja Ganu, Akshay Uttama Nambi, Ni Lao, Gengchen Mai
NeurIPS14
2024 Img2Loc: Revisiting Image Geolocalization using Multi-modality Foundation Models and Image-based Retrieval-Augmented Generation
abstract
Geolocating precise locations from images presents a challenging problem in computer vision and information retrieval. Traditional methods typically employ either classification-dividing the Earth's surface into grid cells and classifying images accordingly, or retrieval-identifying locations by matching images with a database of image-location pairs. However, classification-based approaches are limited by the cell size and cannot yield precise predictions, while retrieval-based systems usually suffer from poor search quality and inadequate coverage of the global landscape at varied scale and aggregation levels. To overcome these drawbacks, we present Img2Loc, a novel system that redefines image geolocalization as a text generation task. This is achieved using cutting-edge large multi-modality models (LMMs) like GPT-4V or LLaVA with retrieval augmented generation. Img2Loc first employs CLIP-based representations to generate an image-based coordinate query database. It then uniquely combines query results with images itself, forming elaborate prompts customized for LMMs. When tested on benchmark datasets such as Im2GPS3k and YFCC4k, Img2Loc not only surpasses the performance of previous state-of-the-art models but does so without any model training. A video demonstration of the system can be accessed via this link https://drive.google.com/file/d/16A6A-mc7AyUoKHRH3_WBRToRC13sn7tU/view?usp=sharing
Zhongliang Zhou, Jielu Zhang, Zihan Guan 0001, Mengxuan Hu, Ni Lao, Lan Mu, Sheng Li 0001, Gengchen Mai
SIGIR5
2023 RARR: Researching and Revising What Language Models Say, Using Language Models
abstract
Luyu Gao, Zhuyun Dai, Panupong Pasupat, Anthony Chen, Arun Tejasvi Chaganty, Yicheng Fan, Vincent Zhao, Ni Lao, Hongrae Lee, Da-Cheng Juan, Kelvin Guu. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023.
Luyu Gao, Zhuyun Dai, Panupong Pasupat, Anthony Chen, Arun Tejasvi Chaganty, Yicheng Fan, Vincent Y. Zhao, Ni Lao, Hongrae Lee, Da-Cheng Juan, Kelvin Guu
ACL (1)8
2023 Rethink Geographical Generalizability with Unsupervised Self-Attention Model Ensemble: A Case Study of OpenStreetMap Missing Building Detection in Africa
abstract
The recent advance of adapting pre-trained task-agnostic artificial intelligence (AI) models leads to great successes in downstream tasks via fine-tuning, or low-resource (i.e., few-shot and zero-shot) learning. However, when adapting such pre-trained AI models to geographical applications, it is still challenging to find the "sweet spot" of the model's generalizability and specializability (e.g., geographic generalizability v.s. spatial heterogeneity). For instance, a building detection task may require vision models with different parameters across different geographic areas of the world. In this paper, we rethink this interesting topic, namely Geographical Generalizability of GeoAI models, with a case study of detecting OpenStreetMap (OSM) missing buildings across different countries in sub-Saharan Africa. We consider a real-world scenario, in which we first train a Single-Shot Multibox Detection (SSD) base model for OSM missing building detection in Kakola, Tanzania, where a previous humanitarian mapping project of OSM was organized to map all possible buildings. Then we extrapolate this base model using Few-Shot Transfer Learning (FSTL) to a set of areas in the proximity of the test area in Cameroon. Here, we develop a Geographical Weighted Model Ensemble (GWME) method to improve Geographical Generalizability of GeoAI models. Moreover, we compare four unsupervised model ensemble weighting strategies: 1) Average weighting, 2) Image similarity weighting, 3) Geographical distance weighting, and 4) Self-attention-based weighting. Experiments show promising results of the proposed GWME method, which implicitly generates model weights from their location embedding and image feature embedding in an unsupervised manner. More specifically, the self-attention-based model ensemble achieves the highest performance. The results shed inspiring light on improving the generalizability and replicability of GeoAI models across geographic areas. Data and code are available at https://github.com/tum-bgd/GWME.
Hao Li 0019, Jiapan Wang, Johann Maximilian Zollner, Gengchen Mai, Ni Lao, Martin Werner 0001
SIGSPATIAL/GIS5
2023 CSP: Self-Supervised Contrastive Spatial Pre-Training for Geospatial-Visual Representations
abstract
Geo-tagged images are publicly available in large quantities, whereas labels such as object classes are rather scarce and expensive to collect. Meanwhile, contrastive learning has achieved tremendous success in various natural image and language tasks with limited labeled data. However, existing methods fail to fully leverage geospatial information, which can be paramount to distinguishing objects that are visually similar. To directly leverage the abundant geospatial information associated with images in pre-training, fine-tuning, and inference stages, we present Contrastive Spatial Pre-Training (CSP), a self-supervised learning framework for geo-tagged images. We use a dual-encoder to separately encode the images and their corresponding geo-locations, and use contrastive objectives to learn effective location representations from images, which can be transferred to downstream supervised tasks such as image classification. Experiments show that CSP can improve model performance on both iNat2018 and fMoW datasets. Especially, on iNat2018, CSP significantly boosts the model performance with 10-34% relative improvement with various labeled training data sampling ratios.
Gengchen Mai, Ni Lao, Jiaming Song, Stefano Ermon
ICML2
2023 Towards general-purpose representation learning of polygonal geometries
Gengchen Mai, Chiyu Max Jiang, Rui Zhu 0008, Yao Xuan, Ling Cai 0002, Krzysztof Janowicz, Stefano Ermon, Ni Lao
GeoInformatica9
2023 Geo-knowledge-guided GPT models improve the extraction of location descriptions from disaster-related social media messages
abstract
Social media messages posted by people during natural disasters often contain important location descriptions, such as the locations of victims. Recent research has shown that many of these location descriptions go beyond simple place names, such as city names and street names, and are difficult to extract using typical named entity recognition (NER) tools. While advanced machine learning models could be trained, they require large labeled training datasets that can be time-consuming and labor-intensive to create. In this work, we propose a method that fuses geo-knowledge of location descriptions and a Generative Pre-trained Transformer (GPT) model, such as ChatGPT and GPT-4. The result is a geo-knowledge-guided GPT model that can accurately extract location descriptions from disaster-related social media messages. Also, only 22 training examples encoding geo-knowledge are used in our method. We conduct experiments to compare this method with nine alternative approaches on a dataset of tweets from Hurricane Harvey. Our method demonstrates an over 40% improvement over typically used NER approaches. The experiment results also show that geo-knowledge is indispensable for guiding the behavior of GPT models. The extracted location descriptions can help disaster responders reach victims more quickly and may even save lives.
Yingjie Hu 0001, Gengchen Mai, Chris Cundy, Kristy Choi, Ni Lao, Gaurish Lakhanpal, Ryan Zhenqi Zhou, Kenneth Joseph
Int. J. Geogr. Inf. Sci.5
2022 Towards a foundation model for geospatial artificial intelligence (vision paper)
abstract
Large pre-trained models, also known as foundation models (FMs), are trained in a task-agnostic manner on large-scale data and can be adapted to a wide range of downstream tasks by fine tuning, few-shot, or even zero-shot learning. Despite their successes in language and vision tasks, we have yet to see an attempt to develop foundation models for geospatial artificial intelligence (GeoAI). In this work, we explore the promises and challenges for developing multimodal foundation models for GeoAI. We first show the advantages of this idea by testing the performance of existing Large pre-trained Language Models (LLMs) (e.g. GPT-2 and GPT-3) on two geospatial semantics tasks. Results indicate that these task-agnostic LLMs can outperform task-specific fully-supervised models on both tasks with 2--9% improvement in a few-shot learning setting. However, we also show the limitations of these existing foundation models given the multimodality nature of GeoAI, especially when dealing with geometries in conjunction with other modalities. So we discuss the possibility of a multimodal foundation model which can reason over various types of geospatial data through geospatial alignments. We conclude this paper by discussing the unique risks and challenges to develop such model for GeoAI.
Gengchen Mai, Chris Cundy, Kristy Choi, Yingjie Hu 0001, Ni Lao, Stefano Ermon
SIGSPATIAL/GIS5
2022 A review of location encoding for GeoAI: methods and applications
abstract
A common need for artificial intelligence models in the broader geoscience is to encode various types of spatial data, such as points, polylines, polygons, graphs, or rasters, in a hidden embedding space so that they can be readily incorporated into deep learning models. One fundamental step is to encode a single point location into an embedding space, such that this embedding is learning-friendly for downstream machine learning models. We call this process location encoding. However, there lacks a systematic review on location encoding, its potential applications, and key challenges that need to be addressed. This paper aims to fill this gap. We first provide a formal definition of location encoding, and discuss the necessity of it for GeoAI research. Next, we provide a comprehensive survey about the current landscape of location encoding research. We classify location encoding models into different categories based on their inputs and encoding methods, and compare them based on whether they are parametric, multi-scale, distance preserving, and direction aware. We demonstrate that existing location encoders can be unified under one formulation framework. We also discuss the application of location encoding. Finally, we point out several challenges that need to be solved in the future.
Gengchen Mai, Krzysztof Janowicz, Yingjie Hu 0001, Song Gao 0001, Bo Yan 0003, Rui Zhu 0008, Ling Cai 0002, Ni Lao
Int. J. Geogr. Inf. Sci.8
2020 Multi-Scale Representation Learning for Spatial Feature Distributions using Grid Cells
Gengchen Mai, Krzysztof Janowicz, Bo Yan 0003, Rui Zhu 0008, Ling Cai 0002, Ni Lao
ICLR6
2019 Contextual Graph Attention for Answering Logical Queries over Incomplete Knowledge Graphs
abstract
Recently, several studies have explored methods for using KG embedding to answer logical queries. These approaches either treat embedding learning and query answering as two separated learning tasks, or fail to deal with the variability of contributions from different query paths. We proposed to leverage a graph attention mechanism to handle the unequal contribution of different query paths. However, commonly used graph attention assumes that the center node embedding is provided, which is unavailable in this task since the center node is to be predicted. To solve this problem we propose a multi-head attention-based end-to-end logical query answering model, called Contextual Graph Attention model (CGA), which uses an initial neighborhood aggregation layer to generate the center embedding, and the whole model is trained jointly on the original KG structure as well as the sampled query-answer pairs. We also introduce two new datasets, DB18 and WikiGeo19, which are rather large in size compared to the existing datasets and contain many more relation types, and use them to evaluate the performance of the proposed model. Our result shows that the proposed CGA with fewer learnable parameters consistently outperforms the baseline models on both datasets as well as Bio dataset.
Gengchen Mai, Krzysztof Janowicz, Bo Yan 0003, Rui Zhu 0008, Ling Cai 0002, Ni Lao
K-CAP6
2018 Memory Augmented Policy Optimization for Program Synthesis and Semantic Parsing
abstract
We present Memory Augmented Policy Optimization (MAPO), a simple and novel way to leverage a memory buffer of promising trajectories to reduce the variance of policy gradient estimate. MAPO is applicable to deterministic environments with discrete actions, such as structured prediction and combinatorial optimization tasks. We express the expected return objective as a weighted sum of two terms: an expectation over the high-reward trajectories inside the memory buffer, and a separate expectation over trajectories outside the buffer. To make an efficient algorithm of MAPO, we propose: (1) memory weight clipping to accelerate and stabilize training; (2) systematic exploration to discover high-reward trajectories; (3) distributed sampling from inside and outside of the memory buffer to scale up training. MAPO improves the sample efficiency and robustness of policy gradient, especially on tasks with sparse rewards. We evaluate MAPO on weakly supervised program synthesis from natural language (semantic parsing). On the WikiTableQuestions benchmark, we improve the state-of-the-art by 2.6%, achieving an accuracy of 46.3%. On the WikiSQL benchmark, MAPO achieves an accuracy of 74.9% with only weak supervision, outperforming several strong baselines with full supervision. Our source code is available at https://goo.gl/TXBp4e
Mohammad Norouzi 0002, Jonathan Berant, Quoc V. Le, Ni Lao
NeurIPS5
2017 Neural Symbolic Machines: Learning Semantic Parsers on Freebase with Weak Supervision
abstract
Harnessing the statistical power of neural networks to perform language understanding and symbolic reasoning is difficult, when it requires executing efficient discrete operations against a large knowledge-base.In this work, we introduce a Neural Symbolic Machine (NSM), which contains (a) a neural "programmer", i.e., a sequence-to-sequence model that maps language utterances to programs and utilizes a key-variable memory to handle compositionality (b) a symbolic "computer", i.e., a Lisp interpreter that performs program execution, and helps find good programs by pruning the search space.We apply REINFORCE to directly optimize the task reward of this structured prediction problem.To train with weak supervision and improve the stability of REINFORCE we augment it with an iterative maximum-likelihood training process.NSM outperforms the state-of-theart on the WEBQUESTIONSSP dataset when trained from question-answer pairs only, without requiring any feature engineering or domain-specific knowledge.
Jonathan Berant, Quoc V. Le, Kenneth D. Forbus, Ni Lao
ACL (1)5
2015 Never-Ending Learning
abstract
Whereas people learn many different types of knowledge from diverse experiences over many years, most current machine learning systems acquire just a single function or data model from just a single data set. We propose a never-ending learning paradigm for machine learning, to better reflect the more ambitious and encompassing type of learning performed by humans. As a case study, we describe the Never-Ending Language Learner (NELL), which achieves some of the desired properties of a never-ending learner, and we discuss lessons learned. NELL has been learning to read the web 24 hours/day since January 2010, and so far has acquired a knowledge base with over 80 million confidence-weighted beliefs (e.g., servedWith(tea, biscuits)). NELL has also learned millions of features and parameters that enable it to read these beliefs from the web. Additionally, it has learned to reason over these beliefs to infer new beliefs, and is able to extend its ontology by synthesizing new relational predicates. NELL can be tracked online at http://rtw.ml.cmu.edu, and followed on Twitter at @CMUNELL.
Tom M. Mitchell, William W. Cohen, Estevam Hruschka, Partha P. Talukdar, Justin Betteridge, Andrew Carlson, Bhavana Dalvi, Matt Gardner 0001, Bryan Kisiel, Jayant Krishnamurthy, Ni Lao, Kathryn Mazaitis, Thahir Mohamed, Ndapandula Nakashole, Emmanouil A. Platanios, Alan Ritter, Mehdi Samadi, Burr Settles, Richard C. Wang, Derry Wijaya, Abhinav Gupta 0001, Xinlei Chen, Abulhair Saparov, Malcolm Greaves, Joel Welling
AAAI11
2015 Learning Relational Features with Backward Random Walks
abstract
Ni Lao, Einat Minkov, William Cohen. Proceedings of the 53rd Annual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2015.
Ni Lao, Einat Minkov, William W. Cohen
ACL (1)1
2015 Efficient inference and learning in a large knowledge base - Reasoning with extracted information using a locally groundable first-order probabilistic logic
William Yang Wang, Kathryn Mazaitis, Ni Lao, William W. Cohen
Mach. Learn.3
2014 Knowledge vault: a web-scale approach to probabilistic knowledge fusion
abstract
Recent years have witnessed a proliferation of large-scale knowledge bases, including Wikipedia, Freebase, YAGO, Microsoft's Satori, and Google's Knowledge Graph. To increase the scale even further, we need to explore automatic methods for constructing knowledge bases. Previous approaches have primarily focused on text-based extraction, which can be very noisy. Here we introduce Knowledge Vault, a Web-scale probabilistic knowledge base that combines extractions from Web content (obtained via analysis of text, tabular data, page structure, and human annotations) with prior knowledge derived from existing knowledge repositories. We employ supervised machine learning methods for fusing these distinct information sources. The Knowledge Vault is substantially bigger than any previously published structured knowledge repository, and features a probabilistic inference system that computes calibrated probabilities of fact correctness. We report the results of multiple studies that explore the relative utility of the different information sources and extraction methods.
Xin Dong 0001, Evgeniy Gabrilovich, Geremy Heitz, Wilko Horn, Ni Lao, Kevin Murphy 0002, Thomas Strohmann, Shaohua Sun, Wei Zhang 0152
KDD5
2012 Reading The Web with Learned Syntactic-Semantic Inference Rules
Ni Lao, Amarnag Subramanya, Fernando Pereira 0003, William W. Cohen
EMNLP-CoNLL1
2011 Random Walk Inference and Learning in A Large Scale Knowledge Base
Ni Lao, Tom M. Mitchell, William W. Cohen
EMNLP1
2011 Conditional topical coding: an efficient topic model conditioned on rich features
abstract
Probabilistic topic models have shown remarkable success in many application domains. However, a probabilistic conditional topic model can be extremely inefficient when considering a rich set of features because it needs to define a normalized distribution, which usually involves a hard-to-compute partition function. This paper presents conditional topical coding (CTC), a novel formulation of conditional topic models which is non-probabilistic. CTC relaxes the normalization constraints as in probabilistic models and learns non-negative document codes and word codes. CTC does not need to define a normalized distribution and can efficiently incorporate a rich set of features for improved topic discovery and prediction tasks. Moreover, CTC can directly control the sparsity of inferred representations by using appropriate regularization. We develop an efficient and easy-to-implement coordinate descent learning algorithm, of which each coding substep has a closed-form solution. Finally, we demonstrate the advantages of CTC on online review analysis datasets. Our results show that conditional topical coding can achieve state-of-the-art prediction performance and is much more efficient in training (one order of magnitude faster) and testing (two orders of magnitude faster) than probabilistic conditional topic models.
Jun Zhu 0001, Ni Lao, Ning Chen 0002, Eric P. Xing
KDD2
2010 Fast query execution for retrieval models based on path-constrained random walks
abstract
Many recommendation and retrieval tasks can be represented as proximity queries on a labeled directed graph, with typed nodes representing documents, terms, and metadata, and labeled edges representing the relationships between them. Recent work has shown that the accuracy of the widely-used random-walk-based proximity measures can be improved by supervised learning - in particular, one especially effective learning technique is based on Path-Constrained Random Walks (PCRW), in which similarity is defined by a learned combination of constrained random walkers, each constrained to follow only a particular sequence of edge labels away from the query nodes. The PCRW based method significantly outperformed unsupervised random walk based queries, and models with learned edge weights. Unfortunately, PCRW query systems are expensive to evaluate. In this study we evaluate the use of approximations to the computation of the PCRW distributions, including fingerprinting, particle filtering, and truncation strategies. In experiments on several recommendation and retrieval problems using two large scientific publications corpora we show speedups of factors of 2 to 100 with little loss in accuracy.
Ni Lao, William W. Cohen
KDD1
2010 Grafting-light: fast, incremental feature selection and structure learning of Markov random fields
abstract
Feature selection is an important task in order to achieve better generalizability in high dimensional learning, and structure learning of Markov random fields (MRFs) can automatically discover the inherent structures underlying complex data. Both problems can be cast as solving an l1-norm regularized parameter estimation problem. The existing Grafting method can avoid doing inference on dense graphs in structure learning by incrementally selecting new features. However, Grafting performs a greedy step to optimize over free parameters once new features are included. This greedy strategy results in low efficiency when parameter learning is itself non-trivial, such as in MRFs, in which parameter learning depends on an expensive subroutine to calculate gradients. The complexity of calculating gradients in MRFs is typically exponential to the size of maximal cliques.
Jun Zhu 0001, Ni Lao, Eric P. Xing
KDD2
2010 Efficient Relational Learning with Hidden Variable Detection
abstract
Markov networks (MNs) can incorporate arbitrarily complex features in modeling relational data. However, this flexibility comes at a sharp price of training an exponentially complex model. To address this challenge, we propose a novel relational learning approach, which consists of a restricted class of relational MNs (RMNs) called relation tree-based RMN (treeRMN), and an efficient Hidden Variable Detection algorithm called Contrastive Variable Induction (CVI). On one hand, the restricted treeRMN only considers simple (e.g., unary and pairwise) features in relational data and thus achieves computational efficiency; and on the other hand, the CVI algorithm efficiently detects hidden variables which can capture long range dependencies. Therefore, the resultant approach is highly efficient yet does not sacrifice its expressive power. Empirical results on four real datasets show that the proposed relational learning method can achieve similar prediction quality as the state-of-the-art approaches, but is significantly more efficient in training; and the induced hidden variables are semantically meaningful and crucial to improve the training speed and prediction qualities of treeRMNs.
Ni Lao, Jun Zhu 0001, William W. Cohen
NIPS1
2010 Relational retrieval using a combination of path-constrained random walks
Ni Lao, William W. Cohen
Mach. Learn.1
2007 Utility-based information distillation over temporally sequenced documents
abstract
This paper examines a new approach to information distillation over temporally ordered documents, and proposes a novel evaluation scheme for such a framework. It combines the strengths of and extends beyond conventional adaptive filtering, novelty detection and non-redundant passage ranking with respect to long-lasting information needs ("tasks" with multiple queries). Our approach supports fine-grained user feedback via highlighting of arbitrary spans of text, and leverages such information for utility optimization in adaptive settings. For our experiments, we defined hypothetical tasks based on news events in the TDT4 corpus, with multiple queries per task. Answer keys (nuggets) were generated for each query and a semi-automatic procedure was used for acquiring rules that allow automatically matching nuggets against system responses. We also propose an extension of the NDCG metric for assessing the utility of ranked passages as a combination of relevance and novelty. Our results show encouraging utility enhancements using the new approach, compared to the baseline systems without incremental learning or the novelty detection components.
Yiming Yang 0002, Abhimanyu Lad, Ni Lao, Abhay Harpale, Bryan Kisiel, Monica Rogati
SIGIR3
2006 Automated known problem diagnosis with event traces
abstract
Computer problem diagnosis remains a serious challenge to users and support professionals. Traditional troubleshooting methods relying heavily on human intervention make the process inefficient and the results inaccurate even for solved problems, which contribute significantly to user's dissatisfaction. We propose to use system behavior information such as system event traces to build correlations with solved problems, instead of using only vague text descriptions as in existing practices. The goal is to enable automatic identification of the root cause of a problem if it is a known one, which would further lead to its resolution. By applying statistical learning techniques to classifying system call sequences, we show our approach can achieve considerable accuracy of root cause recognition by studying four case examples.
Chun Yuan 0004, Ni Lao, Ji-Rong Wen, Zheng Zhang 0001, Yi-Min Wang, Wei-Ying Ma
EuroSys2
2004 Why PCs Are Fragile and What We Can Do About It: A Study of Windows Registry Problems
abstract
Software configuration problems are a major source of failures in computer systems. In this paper, we present a new framework for categorizing configuration problems. We apply this categorization to Windows registry-related problems obtained from various internal as well as external sources. Although infrequent, registry-related problems are difficult to diagnose and repair. Consequently they frustrate the users. We classify problems based on their manifestation and the scope of impact to gain useful insights into how problems affect users and why PCs are fragile. We then describe techniques to identify and eliminate such registry failures. We propose health predicate monitoring for detecting known problems, fault injection for improving application, robustness, and access protection mechanisms for preventing fragility problems.
Archana Ganapathi, Yi-Min Wang, Ni Lao, Ji-Rong Wen
DSN3
2004 Combining High Level Symptom Descriptions and Low Level State Information for Configuration Fault Diagnosis
Ni Lao, Ji-Rong Wen, Wei-Ying Ma, Yi-Min Wang
LISA1
2004 Probabilistic model for contextual retrieval
abstract
Contextual retrieval is a critical technique for facilitating many important applications such as mobile search, personalized search, PC troubleshooting, etc. Despite of its importance, there is no comprehensive retrieval model to describe the contextual retrieval process. We observed that incompatible context, noisy context and incomplete query are several important issues commonly existing in contextual retrieval applications. However, these issues have not been previously explored and discussed. In this paper, we propose probabilistic models to address these problems. Our study clearly shows that query log is the key to build effective contextual retrieval models. We also conduct a case study in the PC troubleshooting domain to testify the performance of the proposed models and experimental results show that the models can achieve very good retrieval precision.
Ji-Rong Wen, Ni Lao, Wei-Ying Ma
SIGIR2
2003 Technical Solutions of TsinghuAeolus for Robotic Soccer
Jinyi Yao, Ni Lao, Yunpeng Cai, Zengqi Sun
RoboCup2