Deep Reinforcement Learning for Image Hashing
Loading...
Files
Date
2024
Journal Title
Journal ISSN
Volume Title
Publisher
NHCE
Abstract
The rapid increase in digital images necessitates efficient retrieval and storage methods, where traditional hashing techniques struggle with scalability and accuracy. This project introduces Deep Reinforcement Learning (DRL) for Image Hashing (DRLIH), leveraging DRL's sequential decision-making to enhance hashing function learning and retrieval performance. Traditional hashing methods like Locality-Sensitive Hashing (LSH) and spectral hashing treat hashing functions independently, ignoring correlations between them and leading to suboptimal accuracy. Deep learning methods using convolutional neural networks (CNNs) for hash code generation also lack mechanisms to refine hashing functions sequentially based on previous errors. DRLIH models the hashing process as a Markov Decision Process (MDP), incorporating state representation from recurrent neural networks (RNNs) and CNNs, an action space of possible hashing functions, and a reward function providing feedback on hash code accuracy. The DRL agent, which features an RNN-based architecture with Long Short-Term Memory (LSTM) units, is trained using the policy gradient method to maximize cumulative rewards through iterative interactions. By iteratively correcting errors and leveraging historical context, DRLIH offers a robust, scalable solution for large-scale image retrieval, setting a new standard in the field. Future work will explore advanced RL techniques to further enhance performance.