fani-lab / fani-lab/SEERa

ECIR2021.SQE-GAN: A Supervised Query Expansion Scheme via GAN

Open
#32 1 comment 0 reactions 1 assignee Claimed by @soroush-ziaeinejad View on GitHub
literature-review
Dominant language
Jupyter Notebook
Stars
5
Forks
5
PR merge metrics
No merged PRs in 30d

Description

**Why did I choose this paper?** Because this paper uses GAN for the task of query expansion which is one of the IR tasks that is strongly related to my research.

### Main problem:
The main problem of this paper is to find a solution to make the existing methods for query expansion faster and more accurate. Query Expansion (QE) is defined as adding new terms to an input query by a user to make it more precise in order to fulfill the users' needs.
### Existing work:
Existing works on the QE can be divided into two categories:
- **Unsupervised Query Expansion** (UQE)
- **Example**: Many classical algorithms
- Probability models
- Relevance-based language models
- **Disadvantage**: noisy or even harmful
- **Supervised Query Expansion** (SQE)
- **Example**: state-of-the-art in the QE literature
- random walk
- term dependency-based approach
- boosting approach
- Learning-based approaches
- **Disadvantage**: higher response time (because of the feature extraction phase)

### Inputs:

- A set of **original queries** {q1, ..., qn},
- A set of **expanded terms** {t1, ..., tM }

### Outputs:

- top k relevant terms to query q from the candidate ones.

### Method:
**Idea:**
- **Response time**: word embedding can be used to avoid the feature extraction phase (time-consuming part) in SQE
- **Performance**:Deep learning can be used to encode the correlation between an arbitrary pair of query and expanded term

**Steps:**
- Using UQE to get expanded terms
- word embedding technique is used to transform the terms into vectors
- Using GAN, both generative and discriminative models iteratively optimize each other
- Generator re-ranks the expanded terms
- Discriminator calculates the score for the new ranking

### Experimental Setup:
**Dataset**:
- TREC Robust 2004
- 528,000 high-quality documents
- 250 queries for experiments

**Preprocessing**:

- **Stemming**: (Porter)
- **Stopwords removal**: (standard InQuery)
- **Word embedding**: Word2Vec’s Continuous Bag-of-Words (CBOW) (d=100)
- **Initial expanded terms**: 100 terms generated by UQE
- **Final selected terms**: 20 terms (of 100 initial terms)
- **40%** train, **10%** validate, **50%** test
- **Basic retrieval model**: TFIDF

**Metrics**:

- MAP (Mean Average Precision)
- Precision@k (k = [5, 10])
- NDCG

### Baselines:
- **state-of-the-art SQE scheme, SQE-TFS** response time
- **traditional UQE, KL divergence** E2
- **UQE** retrieval effect
- **RankSVM** retrieval effect
- **Deep NN** retrieval effect
- **Sequence to sequence learning** retrieval effect
- **BiLSTM**
- **Query-to-Term Attention**

### Results:
The main contribution of this paper is to propose a fast and high-performance model for QE problem. Results show that the major contribution is in the response time (37% improvement) by removing the feature extraction phase and adding a word-embedding module instead. In addition, SQL-GAN also improves the result compared with the latest deep learning-based QE solutions.
### Code:
The code of this paper is unavailable.
### Presentation:
There is no available presentation for this paper.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.