Henryy-rs / top-k-Ranking-Loss

UCF-Crime-Anomaly-Detection
3 stars 0 forks source link
3d-convolutional-network anomaly-detection

UCF-Crime-Anomaly-Detection

Repository for report submission.

3dResNet Features(depth=152, dim=2048) -> https://drive.google.com/file/d/17wdy_DS9UY37J9XTV5XCLqxOFgXiv3ZK/view?usp=sharing

0. Usage

https://github.com/ekosman/AnomalyDetectionCVPR2018-Pytorch

1. Overview

코로나 19가 장기간 유행하고 언택트 트렌드가 확산되면서 다양한 업종의 무인매장이 들어서고 있다. 이에 따라 무인매장에서 도난을 비롯한 사건사고에 관한 소식이 끊이지 않고 있다. 무인매장에 설치되어 있는 CCTV를 활용하는 이상 탐지 모델을 개발 한다면 도난사고를 예방하고 이상 상황 발생 시 신속하게 대처 할 수 있는 보안 솔루션으로 사용될 수 있을 것이다.

2. Dataset

UCF Crime Dataset https://www.crcv.ucf.edu/projects/real-world/ image

3. Baseline Model

Real-world Anomaly Detection in Surveillance Videos https://arxiv.org/abs/1801.04264 image

4. Changes

4.1. Loss Function

기존 loss function(1)은 다음과 같다.

loss_function

위의 loss function은 hinge-loss function이다. B는 영상의 조각을 담고 있는 가방(Bag), V는 영상의 특징 추출된 값, f는 모델을 의미한다. 이 loss function을 다음과 같이 바꾸었다.

CodeCogsEqn (11)

4.2. Featuere Extraction(C3D, I3D, R3D)

Baseline 모델은 Sports-m1 데이터셋에서 사전 학습된 C3D 모델로 특징추출을 했다. 더 깊은 모델과 방대한 데이터셋으로 학습된 사전 학습 모델을 이용하면 성능을 높일 수 있을 것이라 생각하 여 Kinetics-700, Moments in Time 데이터셋을 이용하여 학습된 3D ResNet 모델로 특징을 추출했다.

Hirokatsu Kataoka, Tenga Wakamiya, Kensho Hara, and Yutaka Satoh, "Would Mega-scale Datasets Further Enhance Spatiotemporal 3D CNNs", arXiv preprint, arXiv:2004.04968, 2020.

the paper: https://arxiv.org/abs/2004.04968

git: https://github.com/kenshohara/3D-ResNets-PyTorch

5. Result

image

R3D + top-k Ranking Loss Models

compare_k

6. Conclusion

사전학습모델을 개선하는 것으로 모델의 성능을 높일 수 있었다. 또한 적절한 k값을 찾아 top-k Ranking Loss를 사용하는 것이 모델의 성능을 향상시킬 수 있음을 확인했다. 다만 동일한 데이터셋에서 더 많은 영상 조각을 학습에 사용하려는 의도로 top-k ranking loss를 고안하였고, 데이터셋마다 적절한 k가 존재할 것이라 기대했지만, 실험 결과 같은 데이터셋에서 특징 추출 기법마다 적절한 k값이 달라지는 이유에 대해서는 알지 못한다. 추후 연구를 통해 UCF-Crime 외에 다른 데이터셋에서의 추가적인 검증이 필요하다.

7. Reference

Training and test codes are borrowed from ekosman. https://github.com/ekosman/AnomalyDetectionCVPR2018-Pytorch